Searched defs:input_arr (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | dot_operation_test.cc | 1687 Array3D<float> input_arr(2, 3, 2); in XLA_TEST_F() local
1704 Array3D<float> input_arr(2, 3, 2); in XLA_TEST_F() local
1725 Array4D<float> input_arr(2, 2, 3, 4); in XLA_TEST_F() local
1742 Array3D<float> input_arr(2, 6, 2); in XLA_TEST_F() local
1766 Array4D<float> input_arr(2, 2, 3, 5); in XLA_TEST_F() local
1831 Array3D<float> input_arr(d0, d1, d2); in DOT_ReorderContracting() local
|
| H A D | exhaustive_unary_test_f64.cc | 46 absl::Span<double> input_arr = (*input_literal)[0].data<double>(); in FillInput() local
|
| H A D | exhaustive_unary_test_complex.cc | 69 absl::Span<NativeT> input_arr = (*input_literal)[0].data<NativeT>(); in FillInput() local
|
| H A D | exhaustive_unary_test_f32_or_smaller.cc | 209 absl::Span<NativeT> input_arr = (*input_literal)[0].data<NativeT>(); in FillInput() local
|
| H A D | reduce_test.cc | 904 std::vector<T> input_arr(num_elems, std::numeric_limits<T>::lowest()); in DoTest() local
|