Searched refs:expected_4d (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | array_elementwise_ops_test.cc | 2835 std::unique_ptr<Array4D<float>> expected_4d(new Array4D<float>(2, 3, 4, 5)); variable
2843 (*expected_4d)(p, z, y, x) = 3.0 * value;
2854 ComputeAndCompareR4<float>(&builder, *expected_4d, {}, error_spec_);
2861 std::unique_ptr<Array4D<float>> expected_4d(new Array4D<float>(2, 3, 4, 5)); in XLA_TEST_F() local
2871 (*expected_4d)(p, z, y, x) = value + operand_b_1d[z]; in XLA_TEST_F()
2882 ComputeAndCompareR4<float>(&builder, *expected_4d, {}, error_spec_); in XLA_TEST_F()
|