Searched refs:s32_arg_shape (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference_test.cc | 715 Shape s32_arg_shape = ShapeUtil::MakeShape(S32, {5, 3}); in TEST_F() local
719 {&f32_arg_shape, &s32_arg_shape, &f32_, &s32_}, {0, 1}, to_apply); in TEST_F()
727 Shape s32_arg_shape = ShapeUtil::MakeShape(S32, {5, 3}); in TEST_F() local
732 {&f32_arg_shape, &s32_arg_shape, &f32_, &s32_}, {0, 1}, to_apply); in TEST_F()
740 Shape s32_arg_shape = ShapeUtil::MakeShape(S32, {5, 3}); in TEST_F() local
744 {&f32_arg_shape, &s32_arg_shape, &f32_, &s32_}, {0, 1}, to_apply); in TEST_F()
763 Shape s32_arg_shape = ShapeUtil::MakeShape(S32, {5, 3}); in TEST_F() local
767 {&f32_arg_shape, &s32_arg_shape, &f32_, &s32_}, {0, 1}, to_apply); in TEST_F()
776 Shape s32_arg_shape = ShapeUtil::MakeShape(S32, {5, 3}); in TEST_F() local
780 {&f32_arg_shape, &s32_arg_shape, &f32_, &s32_}, {0, 1}, to_apply); in TEST_F()
[all …]
|