Searched refs:matrix_32_48_ (Results 1 – 1 of 1) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | shape_inference_test.cc | 47 const Shape matrix_32_48_ = ShapeUtil::MakeShape(F32, {32, 48}); member in xla::__anon8d0540a50111::ShapeInferenceTest 1038 ShapeUtil::Equal(inferred_status_match.ValueOrDie(), matrix_32_48_)) in TEST_F() 1367 {&matrix_32_48_, &matrix_32_64_, &matrix_32_48_}, /*dimension=*/1); in TEST_F() 1410 {&matrix_32_48_, &matrix_32_64_}, /*dimension=*/0); in TEST_F() 1494 {&f32_, &s32_, &pred_, &vector_32_, &matrix_32_48_}, in TEST_F() 1496 {f32_, s32_, pred_, vector_32_, matrix_32_48_}, s32matrix_64_64_)); in TEST_F() 1551 {ShapeUtil::MakeProgramShape({matrix_32_48_}, vector_64_), in TEST_F() 1553 {matrix_32_48_, vector_32_}); in TEST_F() 1560 {ShapeUtil::MakeProgramShape({matrix_32_48_}, vector_32_), in TEST_F() 1562 {matrix_32_48_, tuple_f32_v32}); in TEST_F() [all …]
|