Home
last modified time | relevance | path

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

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