Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/python/compiler/mlir/
H A Dmlir_test.py66 input_data_shapes=["10,10", "10,10"],
83 input_data_shapes=["", ""],
99 input_data_shapes=["10,10", "10,10"],
112 input_data_shapes=["10,11", "10,10"],
/aosp_15_r20/external/tensorflow/tensorflow/python/
H A Dpywrap_mlir.py28 input_data_shapes=None, argument
35 ':'.join(input_data_shapes).encode('utf-8'),
H A Dmlir_wrapper.cc54 const std::string &input_data_shapes, in PYBIND11_MODULE()
60 input_data_types, input_data_shapes, output_names, status.get()); in PYBIND11_MODULE()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/python/
H A Dmlir.cc202 absl::string_view input_data_shapes, in ImportGraphDef() argument
206 auto s = ParseInputArrayInfo(input_names, input_data_types, input_data_shapes, in ImportGraphDef()
H A Dmlir.h54 absl::string_view(input_data_shapes),