Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/jit/
Dtest_util.cc25 std::map<string, std::vector<PartialTensorShape>> expected_shapes) { in ShapeAnnotationsMatch()
/external/tensorflow/tensorflow/compiler/xla/service/
Dreduce_decomposer.cc80 std::vector<Shape> expected_shapes(reduce->input_count()); in HandleReduce() local
Dhlo_creation_utils.cc459 std::vector<Shape> expected_shapes; in MakeReduceHlo() local
/external/tensorflow/tensorflow/python/data/kernel_tests/
Dfrom_tensors_test.py186 def testNestedStructure(self, components, expected_shapes): argument
Dtest_base.py169 expected_shapes=None, argument