/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/ |
H A D | concat_test.cc | 146 auto a = ConstantR2FromArray2D(&builder, Array2D<float>(0, 0)); in XLA_TEST_F() 147 auto b = ConstantR2FromArray2D(&builder, Array2D<float>(0, 0)); in XLA_TEST_F() 159 auto a = ConstantR2FromArray2D(&builder, *a_array); in XLA_TEST_F() 160 auto b = ConstantR2FromArray2D(&builder, *b_array); in XLA_TEST_F() 174 auto a = ConstantR2FromArray2D(&builder, *a_array); in XLA_TEST_F() 175 auto b = ConstantR2FromArray2D(&builder, *b_array); in XLA_TEST_F() 187 auto a = ConstantR2FromArray2D(&builder, Array2D<float>(2, 0)); in XLA_TEST_F() 188 auto b = ConstantR2FromArray2D(&builder, *b_array); in XLA_TEST_F() 198 auto a = ConstantR2FromArray2D(&builder, *a_array); in XLA_TEST_F() 199 auto b = ConstantR2FromArray2D(&builder, *b_array); in XLA_TEST_F() [all …]
|
H A D | dot_operation_test.cc | 114 auto lhs = ConstantR2FromArray2D<T>(&builder, {{3.0f, 4.0f}}); in XLA_TYPED_TEST() 151 auto lhs = ConstantR2FromArray2D<T>(&builder, Array2D<T>(0, 2)); in XLA_TYPED_TEST() 152 auto rhs = ConstantR2FromArray2D<T>(&builder, Array2D<T>(2, 0)); in XLA_TYPED_TEST() 162 auto lhs = ConstantR2FromArray2D<T>(&builder, Array2D<T>(0, 2)); in XLA_TYPED_TEST() 163 auto rhs = ConstantR2FromArray2D<T>( in XLA_TYPED_TEST() 174 auto lhs = ConstantR2FromArray2D<T>( in XLA_TYPED_TEST() 176 auto rhs = ConstantR2FromArray2D<T>(&builder, Array2D<T>(2, 0)); in XLA_TYPED_TEST() 186 auto lhs = ConstantR2FromArray2D<T>(&builder, Array2D<T>(2, 0)); in XLA_TYPED_TEST() 187 auto rhs = ConstantR2FromArray2D<T>(&builder, Array2D<T>(0, 2)); in XLA_TYPED_TEST() 574 ConstantR2FromArray2D<T>(&builder, {{1.0f, 2.0f}, {3.0f, 4.0f}}); in XLA_TYPED_TEST() [all …]
|
H A D | binop_scaling_test.cc | 36 auto lhs = ConstantR2FromArray2D<float>(&builder, *alhs); in TEST_F() 37 auto rhs = ConstantR2FromArray2D<float>(&builder, *arhs); in TEST_F() 52 auto lhs = ConstantR2FromArray2D<float>(&builder, *alhs); in TEST_F() 53 auto rhs = ConstantR2FromArray2D<float>(&builder, *arhs); in TEST_F() 68 auto lhs = ConstantR2FromArray2D<float>(&builder, *alhs); in TEST_F() 69 auto rhs = ConstantR2FromArray2D<float>(&builder, *arhs); in TEST_F() 84 auto lhs = ConstantR2FromArray2D<float>(&builder, *alhs); in TEST_F() 85 auto rhs = ConstantR2FromArray2D<float>(&builder, *arhs); in TEST_F()
|
H A D | transpose_test.cc | 41 auto lhs = ConstantR2FromArray2D<float>(&builder, Array2D<float>(0, 0)); in XLA_TEST_F() 49 auto lhs = ConstantR2FromArray2D<float>(&builder, Array2D<float>(0, 42)); in XLA_TEST_F() 57 auto lhs = ConstantR2FromArray2D<float>(&builder, Array2D<float>(7, 0)); in XLA_TEST_F() 124 auto computed = ConstantR2FromArray2D<float>(&builder, input); in TEST_F() 138 auto operand = ConstantR2FromArray2D<float>(&builder, *aoperand); in TEST_F() 150 auto operand = ConstantR2FromArray2D<float>(&builder, *aoperand); in TEST_F()
|
H A D | matrix_ops_simple_test.cc | 58 auto data = ConstantR2FromArray2D<T>(&builder, { in XLA_TYPED_TEST() 87 auto data = ConstantR2FromArray2D<T>(&builder, { in XLA_TYPED_TEST() 102 auto lhs = ConstantR2FromArray2D<T>(&builder, { in XLA_TYPED_TEST() 106 auto rhs = ConstantR2FromArray2D<T>(&builder, { in XLA_TYPED_TEST() 138 auto lhs = ConstantR2FromArray2D<T>(&builder, *alhs); in TestImpl() 139 auto rhs = ConstantR2FromArray2D<T>(&builder, *arhs); in TestImpl()
|
H A D | slice_test.cc | 81 auto original = ConstantR2FromArray2D<float>(&builder, Array2D<float>(0, 0)); in XLA_TEST_F() 89 auto original = ConstantR2FromArray2D<float>(&builder, Array2D<float>(0, 20)); in XLA_TEST_F() 97 auto original = ConstantR2FromArray2D<float>(&builder, Array2D<float>(3, 0)); in XLA_TEST_F() 112 auto original = ConstantR2FromArray2D<float>(&builder, values); in XLA_TEST_F() 130 auto original = ConstantR2FromArray2D<float>(&builder, values); in TEST_F() 151 auto original = ConstantR2FromArray2D<float>(&builder, values); in TEST_F() 172 auto original = ConstantR2FromArray2D(&builder, values); in TEST_F()
|
H A D | constants_test.cc | 99 ConstantR2FromArray2D<float>(&builder, Array2D<float>(0, 2)); in TEST_F() 109 ConstantR2FromArray2D<float>(&builder, *constant); in TEST_F()
|
H A D | tuple_test.cc | 140 ConstantR2FromArray2D<float>(&builder, Array2D<float>(0, 101))}); in XLA_TEST_F()
|
H A D | reduce_test.cc | 972 auto m = ConstantR2FromArray2D(&builder, initializer); in XLA_TEST_F() 986 auto m = ConstantR2FromArray2D(&builder, initializer); in XLA_TEST_F()
|
H A D | array_elementwise_ops_test.cc | 2064 auto v = ConstantR2FromArray2D<float>(&builder, Array2D<float>(0, 2)); in XLA_TEST_F()
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/lib/ |
H A D | quantize_test.cc | 322 auto x = ConstantR2FromArray2D<uint32_t>(&builder, input_packed); in XLA_TEST_F() 336 auto x = ConstantR2FromArray2D<uint32_t>(&builder, input_packed); in XLA_TEST_F()
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
H A D | image_ops.cc | 392 row_iou, xla::ConstantR2FromArray2D<bool>(builder, {{false}}), in operator ()()
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/ |
H A D | xla_builder.h | 1699 XlaOp ConstantR2FromArray2D(XlaBuilder* builder, 2766 XlaOp ConstantR2FromArray2D(XlaBuilder* builder, in ConstantR2FromArray2D() function
|