Searched refs:operand_shape_ (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference_test.cc | 77 operand_shape_ = ShapeUtil::MakeShape(F32, {8, 16}); in SelectAndScatterShapeInferenceTest()
95 Shape operand_shape_; member in xla::__anon8d0540a50111::SelectAndScatterShapeInferenceTest
329 operand_shape_, select_program_shape_, window_, source_shape_, in TEST_F()
333 ASSERT_TRUE(ShapeUtil::Equal(operand_shape_, inferred)); in TEST_F()
339 operand_shape_, select_program_shape_, window_, source_shape_fail, in TEST_F()
350 operand_shape_, select_program_shape_fail, window_, source_shape_, in TEST_F()
361 operand_shape_, select_program_shape_fail, window_, source_shape_, in TEST_F()
372 operand_shape_, select_program_shape_fail, window_, source_shape_, in TEST_F()
383 operand_shape_, select_program_shape_fail, window_, source_shape_, in TEST_F()
|