Home
last modified time | relevance | path

Searched refs:IsGraphOutput (Results 1 – 6 of 6) sorted by relevance

/external/pytorch/torch/csrc/jit/passes/onnx/
Dnaming.cpp90 bool IsGraphOutput(const Value* v, const std::shared_ptr<Graph>& graph) const;
132 bool NodeNameGenerator::IsGraphOutput( in IsGraphOutput() function in torch::jit::onnx::__anon641fc3f90211::NodeNameGenerator
148 if (!IsGraphOutput(output, graph_)) { in UpdateOutputsNames()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dmodel.h104 bool IsGraphOutput(ValueId id) const;
Dmodel.cc79 bool GraphFloat32::IsGraphOutput(ValueId id) const { in IsGraphOutput() function in tflite::gpu::GraphFloat32
Dgpu_model.cc226 if (!(graph.IsGraphInput(t->id) || graph.IsGraphOutput(t->id))) { in ReserveGraphTensors()
Dmodel_builder.cc3359 if (!graph->IsGraphOutput(value_id)) { in BuildModelEnforceIO()
/external/tensorflow/tensorflow/lite/delegates/gpu/gl/
Dcompiler.cc173 graph.IsGraphInput(value->id) || graph.IsGraphOutput(value->id) || in Compile()