Searched refs:f32_scalar (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | concat_test.cc | 562 auto f32_scalar = ShapeUtil::MakeShape(xla::F32, {}); in XLA_TEST_F() local
569 auto x = Parameter(&builder, 0, f32_scalar, "x"); in XLA_TEST_F()
570 auto y = Parameter(&builder, 1, f32_scalar, "y"); in XLA_TEST_F()
584 auto f32_scalar = ShapeUtil::MakeShape(xla::F32, {}); in XLA_TEST_F() local
594 auto y = Parameter(&builder, 1, f32_scalar, "y"); in XLA_TEST_F()
595 auto z = Parameter(&builder, 2, f32_scalar, "z"); in XLA_TEST_F()
611 auto f32_scalar = ShapeUtil::MakeShape(xla::F32, {}); in XLA_TEST_F() local
622 auto y = Parameter(&builder, 1, f32_scalar, "y"); in XLA_TEST_F()
623 auto z = Parameter(&builder, 2, f32_scalar, "y"); in XLA_TEST_F()
|