Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
H A Dmlir_roundtrip_flags.cc90 std::vector<llvm::Optional<std::vector<int>>> node_shapes; in ParseInputArrayInfo() local
146 const std::vector<llvm::Optional<std::vector<int>>>& node_shapes, in ParseInputArrayInfo()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/python/
H A Dgraphdef_to_tfl_flatbuffer.cc61 std::vector<llvm::Optional<std::vector<int>>> node_shapes; in ConvertGraphDefToTFLiteFlatBuffer() local
H A Dsaved_model_to_tfl_flatbuffer.cc138 std::vector<llvm::Optional<std::vector<int>>> node_shapes; in ConvertSavedModelToTFLiteFlatBuffer() local
H A Djax_to_tfl_flatbuffer.cc134 std::vector<llvm::Optional<std::vector<int>>> node_shapes; in ConvertJaxToTFLiteFlatBuffer() local
H A Dtf_tfl_flatbuffer_helpers.cc211 std::vector<llvm::Optional<std::vector<int>>>* node_shapes, in PopulateQuantizationSpecs()
/aosp_15_r20/external/tensorflow/tensorflow/tools/benchmark/
H A Dbenchmark_model.cc123 std::unordered_map<string, TensorShape>* node_shapes) { in GetOutputShapes()