Searched refs:kOutputIndex (Results 1 – 4 of 4) sorted by relevance
| /external/libtextclassifier/native/tensorflow_models/seq_flow_lite/tflite_ops/ |
| D | layer_norm.cc | 49 const int kOutputIndex = 0; variable
71 TfLiteTensor* output = &context->tensors[node->outputs->data[kOutputIndex]]; in Resize()
296 TfLiteTensor* output = &context->tensors[node->outputs->data[kOutputIndex]]; in Eval()
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_avgpooling_op.cc | 78 const int kOutputIndex = 0; in Compute() local
79 this->AllocateEmptyOutputTensor(context, kOutputIndex, &pool_params, in Compute()
|
| D | mkl_maxpooling_op.cc | 86 const int kOutputIndex = 0; in Compute() local
87 this->AllocateEmptyOutputTensor(context, kOutputIndex, &pool_params, in Compute()
|
| D | mkl_pooling_ops_common.h | 544 const int kOutputIndex, in AllocateEmptyOutputTensor() argument
570 AllocateOutputSetMklShape(context, kOutputIndex, output_tensor, in AllocateEmptyOutputTensor()
|