Home
last modified time | relevance | path

Searched defs:node_shapes (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dmlir_roundtrip_flags.cc58 std::vector<llvm::Optional<std::vector<int>>> node_shapes; in ParseInputArrayInfo() local
68 const std::vector<llvm::Optional<std::vector<int>>>& node_shapes, in ParseInputArrayInfo()
/external/tensorflow/tensorflow/compiler/mlir/lite/python/
Dgraphdef_to_tfl_flatbuffer.cc58 std::vector<llvm::Optional<std::vector<int>>> node_shapes; in ConvertGraphDefToTFLiteFlatBuffer() local
Dsaved_model_to_tfl_flatbuffer.cc131 std::vector<llvm::Optional<std::vector<int>>> node_shapes; in ConvertSavedModelToTFLiteFlatBuffer() local
Dtf_tfl_flatbuffer_helpers.cc198 std::vector<llvm::Optional<std::vector<int>>>* node_shapes, in PopulateQuantizationSpecs()
/external/tensorflow/tensorflow/tools/benchmark/
Dbenchmark_model.cc119 std::unordered_map<string, TensorShape>* node_shapes) { in GetOutputShapes()