Searched defs:input_arr (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | exhaustive_unary_test_f64.cc | 46 absl::Span<double> input_arr = (*input_literal)[0].data<double>(); in FillInput() local
|
| D | exhaustive_unary_test_complex.cc | 69 absl::Span<NativeT> input_arr = (*input_literal)[0].data<NativeT>(); in FillInput() local
|
| D | dot_operation_test.cc | 1686 Array3D<float> input_arr(2, 3, 2); in XLA_TEST_F() local
1703 Array3D<float> input_arr(2, 3, 2); in XLA_TEST_F() local
1724 Array4D<float> input_arr(2, 2, 3, 4); in XLA_TEST_F() local
1741 Array3D<float> input_arr(2, 6, 2); in XLA_TEST_F() local
1765 Array4D<float> input_arr(2, 2, 3, 5); in XLA_TEST_F() local
1830 Array3D<float> input_arr(d0, d1, d2); in DOT_ReorderContracting() local
|
| D | exhaustive_unary_test_f32_or_smaller.cc | 206 absl::Span<NativeT> input_arr = (*input_literal)[0].data<NativeT>(); in FillInput() local
|
| D | reduce_test.cc | 899 std::vector<T> input_arr(num_elems, std::numeric_limits<T>::lowest()); in DoTest() local
|