Home
last modified time | relevance | path

Searched refs:tflite_graph_io (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/gl/
Dcompiler.cc111 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()
Dcompiler.h45 const std::unordered_set<int>& tflite_graph_io, // NOLINT
Dapi.h70 const std::unordered_set<int>& tflite_graph_io, // NOLINT
Dapi.cc381 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/
Dgl_delegate.cc162 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()