Searched refs:source_shape_ (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference_test.cc | 78 source_shape_ = ShapeUtil::MakeShape(F32, {4, 8}); in SelectAndScatterShapeInferenceTest()
96 Shape source_shape_; member in xla::__anon8d0540a50111::SelectAndScatterShapeInferenceTest
329 operand_shape_, select_program_shape_, window_, source_shape_, 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()
|