Searched refs:tflite_graph_io (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/gl/ |
| D | compiler.cc | 108 const std::unordered_set<int>& tflite_graph_io, // NOLINT in Compile() argument
171 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 | 378 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 | 163 std::unordered_set<int> tflite_graph_io; // NOLINT in Prepare() local
178 tflite_graph_io.insert(tensor_index); in Prepare()
217 tflite_graph_io.insert(tensor_index); in Prepare()
268 RETURN_IF_ERROR(Compile(compile_options, graph, tflite_graph_io, *shaders, in Prepare()
|