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