Home
last modified time | relevance | path

Searched defs:expected_output_shapes (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/data/kernel_tests/
H A Diterator_test.py818 expected_output_shapes): argument
846 expected_output_shapes): argument
893 expected_output_shapes): argument
/aosp_15_r20/external/tensorflow/tensorflow/c/eager/parallel_device/
H A Dparallel_device_lib.cc328 std::vector<PartialTensorShape> expected_output_shapes(expected_max_outputs); in Execute() local
414 const std::vector<PartialTensorShape>& expected_output_shapes, in Join()
/aosp_15_r20/external/tensorflow/tensorflow/lite/testing/
H A Dtf_driver.h58 const std::vector<std::pair<string, string>>& expected_output_shapes) in CheckResults()
H A Dtest_runner_test.cc41 const std::vector<std::pair<string, string>>& expected_output_shapes) in CheckResults()
H A Dtflite_driver.cc533 const std::vector<std::pair<string, string>>& expected_output_shapes) { in CheckResults()
/aosp_15_r20/external/tensorflow/tensorflow/core/data/
H A Ddataset_test_base.h471 std::vector<PartialTensorShape> expected_output_shapes; member
494 std::vector<PartialTensorShape> expected_output_shapes; member
H A Ddataset_test_base.cc700 const std::vector<PartialTensorShape>& expected_output_shapes) { in CheckDatasetOutputShapes()
726 const std::vector<PartialTensorShape>& expected_output_shapes) { in CheckIteratorOutputShapes()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dxla_call_module_op.cc55 std::vector<PartialTensorShape> expected_output_shapes; in XlaCallModuleOp() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/python/
H A Dtflite_convert_test.py55 expected_output_shapes=None): argument