Searched refs:tflite_graph_io (Results 1 – 5 of 5) sorted by relevance
/external/tensorflow/tensorflow/lite/delegates/gpu/gl/ |
D | compiler.cc | 111 const std::unordered_set<int>& tflite_graph_io, // NOLINT in Compile() argument 174 tflite_graph_io.find(value->tensor.ref) != tflite_graph_io.end(); in Compile()
|
D | compiler.h | 45 const std::unordered_set<int>& tflite_graph_io, // NOLINT
|
D | api.h | 70 const std::unordered_set<int>& tflite_graph_io, // NOLINT
|
D | api.cc | 381 const std::unordered_set<int>& tflite_graph_io, // NOLINT in Compile() argument 396 model, tflite_graph_io, [&](ShaderCode code) -> absl::Status { in Compile()
|
/external/tensorflow/tensorflow/lite/delegates/gpu/ |
D | gl_delegate.cc | 162 std::unordered_set<int> tflite_graph_io; // NOLINT in Prepare() local 177 tflite_graph_io.insert(tensor_index); in Prepare() 216 tflite_graph_io.insert(tensor_index); in Prepare() 267 RETURN_IF_ERROR(Compile(compile_options, graph, tflite_graph_io, *shaders, in Prepare()
|