| /external/tensorflow/tensorflow/contrib/lite/kernels/ |
| D | reshape.cc | 28 constexpr int kOutputTensor = 0; variable
39 TfLiteTensor* output = GetOutput(context, node, kOutputTensor); in Prepare()
74 TfLiteTensor* output = GetOutput(context, node, kOutputTensor); in Eval()
|
| D | l2norm.cc | 35 constexpr int kOutputTensor = 0; variable
44 TfLiteTensor* output = GetOutput(context, node, kOutputTensor); in Prepare()
63 TfLiteTensor* output = GetOutput(context, node, kOutputTensor); in Eval()
|
| D | local_response_norm.cc | 35 constexpr int kOutputTensor = 0; variable
42 TfLiteTensor* output = GetOutput(context, node, kOutputTensor); in Prepare()
64 TfLiteTensor* output = GetOutput(context, node, kOutputTensor); in Eval()
|
| D | sub.cc | 38 constexpr int kOutputTensor = 0; variable
46 TfLiteTensor* output = GetOutput(context, node, kOutputTensor); in Prepare()
87 TfLiteTensor* output = GetOutput(context, node, kOutputTensor); in Eval()
|
| D | space_to_depth.cc | 36 constexpr int kOutputTensor = 0; variable
46 TfLiteTensor* output = GetOutput(context, node, kOutputTensor); in Prepare()
80 TfLiteTensor* output = GetOutput(context, node, kOutputTensor); in Eval()
|
| D | div.cc | 38 constexpr int kOutputTensor = 0; variable
46 TfLiteTensor* output = GetOutput(context, node, kOutputTensor); in Prepare()
87 TfLiteTensor* output = GetOutput(context, node, kOutputTensor); in Eval()
|
| D | resize_bilinear.cc | 37 constexpr int kOutputTensor = 0; variable
56 TfLiteTensor* output = GetOutput(context, node, kOutputTensor); in Prepare()
82 TfLiteTensor* output = GetOutput(context, node, kOutputTensor); in Eval()
|
| D | gather.cc | 30 constexpr int kOutputTensor = 0; variable
40 TfLiteTensor* output = GetOutput(context, node, kOutputTensor); in Prepare()
86 TfLiteTensor* output = GetOutput(context, node, kOutputTensor); in Eval()
|
| D | basic_rnn.cc | 39 constexpr int kOutputTensor = 1; variable
64 TfLiteTensor* output = &context->tensors[node->outputs->data[kOutputTensor]]; in Prepare()
97 TfLiteTensor* output = &context->tensors[node->outputs->data[kOutputTensor]]; in Eval()
|
| D | unidirectional_sequence_rnn.cc | 39 constexpr int kOutputTensor = 1; variable
69 TfLiteTensor* output = &context->tensors[node->outputs->data[kOutputTensor]]; in Prepare()
103 TfLiteTensor* output = &context->tensors[node->outputs->data[kOutputTensor]]; in Eval()
|
| D | mul.cc | 38 constexpr int kOutputTensor = 0; variable
62 TfLiteTensor* output = GetOutput(context, node, kOutputTensor); in Prepare()
154 TfLiteTensor* output = GetOutput(context, node, kOutputTensor); in Eval()
|
| D | add.cc | 38 constexpr int kOutputTensor = 0; variable
62 TfLiteTensor* output = GetOutput(context, node, kOutputTensor); in Prepare()
169 TfLiteTensor* output = GetOutput(context, node, kOutputTensor); in Eval()
|
| D | depthwise_conv.cc | 43 constexpr int kOutputTensor = 0; variable
91 TfLiteTensor* output = GetOutput(context, node, kOutputTensor); in Prepare()
232 TfLiteTensor* output = GetOutput(context, node, kOutputTensor); in Eval()
|
| D | fully_connected.cc | 63 constexpr int kOutputTensor = 0; variable
90 TfLiteTensor* output = GetOutput(context, node, kOutputTensor); in Prepare()
246 TfLiteTensor* output = GetOutput(context, node, kOutputTensor); in Eval()
|
| D | strided_slice.cc | 39 constexpr int kOutputTensor = 0; variable
48 output = GetOutput(context, node, kOutputTensor); in StridedSliceContext()
|
| D | lstm.cc | 71 constexpr int kOutputTensor = 3; variable
253 TfLiteTensor* output = GetOutput(context, node, kOutputTensor); in Prepare()
350 TfLiteTensor* output = GetOutput(context, node, kOutputTensor); in Eval()
|
| D | unidirectional_sequence_lstm.cc | 71 constexpr int kOutputTensor = 3; variable
254 TfLiteTensor* output = GetOutput(context, node, kOutputTensor); in Prepare()
352 TfLiteTensor* output = GetOutput(context, node, kOutputTensor); in Eval()
|
| /external/tensorflow/tensorflow/contrib/tensorrt/ |
| D | tensorrt_test.cc | 63 const char* kOutputTensor = "output"; variable
82 output->setName(kOutputTensor); in CreateNetwork()
105 const int output_index = engine.getBindingIndex(kOutputTensor); in Execute()
|
| /external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/ |
| D | identify_lstm_merge_inputs.cc | 45 if (!GetMatchingRnnArray(model, src_op->outputs[kOutputTensor], in Run()
72 src_op->outputs[kOutputTensor], src_op->outputs[kCellStateTensor])); in Run()
152 src_op->outputs[kOutputTensor]; in Run()
|
| D | lstm_utils.h | 54 kOutputTensor = 3 enumerator
|
| D | identify_lstm_split_inputs.cc | 149 lstm_cell_op->outputs[kOutputTensor] = in Run()
|