experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

internal code change

Browse Source 
PiperOrigin-RevId: 275195470
Change-Id: I96f98efe9ba6a92247fb9d6aade86e6b107c87f9
This commit is contained in:
Taehee Jeong 2019-10-16 23:58:40 -07:00 committed by TensorFlower Gardener
parent 08cc5a44f3
commit cd7458379f
1 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
tensorflow/lite/kernels/kernel_util.h
View File

@ -28,29 +28,29 @@ inline int NumDimensions(const TfLiteTensor* t) { return t->dims->size; }
inline int SizeOfDimension(const TfLiteTensor* t, int dim) { inline int SizeOfDimension(const TfLiteTensor* t, int dim) {
return t->dims->data[dim]; return t->dims->data[dim];
} }
inline const TfLiteTensor* GetInput(TfLiteContext* context, TfLiteNode* node, inline const TfLiteTensor* GetInput(TfLiteContext* context,
int index) { const TfLiteNode* node, int index) {
return &context return &context
->tensors[flatbuffers::EndianScalar(node->inputs->data[index])]; ->tensors[flatbuffers::EndianScalar(node->inputs->data[index])];
} }
inline TfLiteTensor* GetVariableInput(TfLiteContext* context, TfLiteNode* node, inline TfLiteTensor* GetVariableInput(TfLiteContext* context,
int index) { const TfLiteNode* node, int index) {
TfLiteTensor* tensor = TfLiteTensor* tensor =
&context->tensors[flatbuffers::EndianScalar(node->inputs->data[index])]; &context->tensors[flatbuffers::EndianScalar(node->inputs->data[index])];
return (tensor->is_variable) ? tensor : nullptr; return (tensor->is_variable) ? tensor : nullptr;
} }
inline TfLiteTensor* GetOutput(TfLiteContext* context, TfLiteNode* node, inline TfLiteTensor* GetOutput(TfLiteContext* context, const TfLiteNode* node,
int index) { int index) {
return &context return &context
->tensors[flatbuffers::EndianScalar(node->outputs->data[index])]; ->tensors[flatbuffers::EndianScalar(node->outputs->data[index])];
} }
inline TfLiteTensor* GetTemporary(TfLiteContext* context, TfLiteNode* node, inline TfLiteTensor* GetTemporary(TfLiteContext* context,
int index) { const TfLiteNode* node, int index) {
return &context->tensors[flatbuffers::EndianScalar( return &context->tensors[flatbuffers::EndianScalar(
node->temporaries->data[index])]; node->temporaries->data[index])];
} }
inline const TfLiteTensor* GetIntermediates(TfLiteContext* context, inline const TfLiteTensor* GetIntermediates(TfLiteContext* context,
TfLiteNode* node, int index) { const TfLiteNode* node, int index) {
return &context->tensors[node->intermediates->data[index]]; return &context->tensors[node->intermediates->data[index]];
} }
inline int NumInputs(const TfLiteNode* node) { return node->inputs->size; } inline int NumInputs(const TfLiteNode* node) { return node->inputs->size; }