Searched refs:graph_outputs (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/transformations/ |
| D | remove_noop.cc | 113 const auto& graph_outputs = graph->outputs(); in ApplyToNode() local
114 if (std::find(graph_outputs.begin(), graph_outputs.end(), output) != in ApplyToNode()
115 graph_outputs.end()) { in ApplyToNode()
|
| /external/XNNPACK/bench/ |
| D | prelu.cc | 161 const int32_t graph_outputs[1] = { 2 }; in tflite_prelu_f32() local
166 builder.CreateVector<int32_t>(graph_outputs, 1), in tflite_prelu_f32()
|
| D | softmax.cc | 207 const int32_t graph_outputs[1] = { 1 }; in tflite_softmax_f32() local
212 builder.CreateVector<int32_t>(graph_outputs, 1), in tflite_softmax_f32()
|
| D | sigmoid.cc | 204 const int32_t graph_outputs[1] = { 1 }; in tflite_sigmoid_f32() local
209 builder.CreateVector<int32_t>(graph_outputs, 1), in tflite_sigmoid_f32()
|
| D | deconvolution.cc | 380 const int32_t graph_outputs[1] = { 3 }; in tflite_deconvolution_f32() local
385 builder.CreateVector<int32_t>(graph_outputs, 1), in tflite_deconvolution_f32()
|
| D | convolution.cc | 412 const int32_t graph_outputs[1] = { 3 }; in tflite_convolution_f32() local
417 builder.CreateVector<int32_t>(graph_outputs, 1), in tflite_convolution_f32()
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | quantize_nodes.cc | 264 std::set<string> graph_outputs; in RemoveRedundantQuantizations() local
266 graph_outputs.insert(NodeNameFromInput(output_name)); in RemoveRedundantQuantizations()
279 [&inputs_to_rename, &graph_outputs](const NodeMatch& match, in RemoveRedundantQuantizations()
296 graph_outputs.count(dequantize_node.name())) { in RemoveRedundantQuantizations()
|