Home
last modified time | relevance | path

Searched refs:graph_outputs (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/transformations/
Dremove_noop.cc126 const auto& graph_outputs = graph->outputs(); in ApplyToNode() local
127 if (std::find(graph_outputs.begin(), graph_outputs.end(), output) != in ApplyToNode()
128 graph_outputs.end()) { in ApplyToNode()
/external/tensorflow/tensorflow/c/eager/
Dc_api_unified_experimental_graph.cc391 std::vector<TF_Output> graph_outputs; in Finalize() local
392 graph_outputs.reserve(outputs->outputs.size()); in Finalize()
400 graph_outputs.push_back(output->output_); in Finalize()
406 graph_outputs.size(), graph_outputs.data(), in Finalize()
/external/XNNPACK/bench/
Delu.cc142 const std::array<int32_t, 1> graph_outputs{{ 1 }}; in tflite_elu_f32() local
147 builder.CreateVector<int32_t>(graph_outputs.data(), graph_outputs.size()), in tflite_elu_f32()
Dbankers-rounding.cc142 const std::array<int32_t, 1> graph_outputs{{ 1 }}; in tflite_bankers_rounding_f32() local
147 builder.CreateVector<int32_t>(graph_outputs.data(), graph_outputs.size()), in tflite_bankers_rounding_f32()
Dfloor.cc142 const std::array<int32_t, 1> graph_outputs{{ 1 }}; in tflite_floor_f32() local
147 builder.CreateVector<int32_t>(graph_outputs.data(), graph_outputs.size()), in tflite_floor_f32()
Dsquare-root.cc142 const std::array<int32_t, 1> graph_outputs{{ 1 }}; in tflite_square_root_f32() local
147 builder.CreateVector<int32_t>(graph_outputs.data(), graph_outputs.size()), in tflite_square_root_f32()
Dceiling.cc142 const std::array<int32_t, 1> graph_outputs{{ 1 }}; in tflite_ceiling_f32() local
147 builder.CreateVector<int32_t>(graph_outputs.data(), graph_outputs.size()), in tflite_ceiling_f32()
Dhardswish.cc144 const std::array<int32_t, 1> graph_outputs{{ 1 }}; in tflite_hardswish_f32() local
149 builder.CreateVector<int32_t>(graph_outputs.data(), graph_outputs.size()), in tflite_hardswish_f32()
Dsigmoid.cc217 const std::array<int32_t, 1> graph_outputs{{ 1 }}; in tflite_sigmoid_f32() local
222 builder.CreateVector<int32_t>(graph_outputs.data(), graph_outputs.size()), in tflite_sigmoid_f32()
Dprelu.cc163 const int32_t graph_outputs[1] = { 2 }; in tflite_prelu_f32() local
168 builder.CreateVector<int32_t>(graph_outputs, 1), in tflite_prelu_f32()
Dsoftmax.cc214 const int32_t graph_outputs[1] = { 1 }; in tflite_softmax_f32() local
219 builder.CreateVector<int32_t>(graph_outputs, 1), in tflite_softmax_f32()
Daverage-pooling.cc265 const int32_t graph_outputs[1] = { 1 }; in tflite_average_pooling_f32() local
270 builder.CreateVector<int32_t>(graph_outputs, 1), in tflite_average_pooling_f32()
Ddeconvolution.cc380 const int32_t graph_outputs[1] = { 3 }; in tflite_deconvolution_f32() local
385 builder.CreateVector<int32_t>(graph_outputs, 1), in tflite_deconvolution_f32()
Dconvolution.cc654 const int32_t graph_outputs[1] = { 3 }; in tflite_convolution_f32() local
659 builder.CreateVector<int32_t>(graph_outputs, 1), in tflite_convolution_f32()
/external/tensorflow/tensorflow/tools/graph_transforms/
Dquantize_nodes.cc264 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()