/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/ |
H A D | convolution_variants_test.cc | 57 auto input = ConstantR4FromArray4D<float>(&builder, input_array); in XLA_TEST_F() 60 auto filter = ConstantR4FromArray4D<float>(&builder, filter_array); in XLA_TEST_F() 72 auto input = ConstantR4FromArray4D<float>(&builder, input_array); in XLA_TEST_F() 75 auto filter = ConstantR4FromArray4D<float>(&builder, filter_array); in XLA_TEST_F() 88 auto input = ConstantR4FromArray4D<float>(&builder, input_array); in XLA_TEST_F() 91 auto filter = ConstantR4FromArray4D<float>(&builder, filter_array); in XLA_TEST_F() 104 auto input = ConstantR4FromArray4D<float>(&builder, input_array); in XLA_TEST_F() 107 auto filter = ConstantR4FromArray4D<float>(&builder, filter_array); in XLA_TEST_F() 119 auto input = ConstantR4FromArray4D<float>(&builder, input_array); in XLA_TEST_F() 122 auto filter = ConstantR4FromArray4D<float>(&builder, filter_array); in XLA_TEST_F() [all …]
|
H A D | select_and_scatter_test.cc | 365 auto operand = ConstantR4FromArray4D(&builder_, o); in TEST_F() 370 auto source = ConstantR4FromArray4D(&builder_, s); in TEST_F() 390 auto operand = ConstantR4FromArray4D(&builder_, o); in TEST_F() 395 auto source = ConstantR4FromArray4D(&builder_, s); in TEST_F() 415 auto operand = ConstantR4FromArray4D(&builder_, o); in TEST_F() 420 auto source = ConstantR4FromArray4D(&builder_, s); in TEST_F() 437 auto operand = ConstantR4FromArray4D(&builder_, o); in TEST_F() 441 auto source = ConstantR4FromArray4D(&builder_, s); in TEST_F() 464 auto operand = ConstantR4FromArray4D(&builder_, o); in XLA_TEST_F() 469 auto source = ConstantR4FromArray4D(&builder_, s); in XLA_TEST_F() [all …]
|
H A D | bfloat16_test.cc | 83 auto operand = ConstantR4FromArray4D<bfloat16>( in XLA_TEST_F() 119 auto operand = ConstantR4FromArray4D<bfloat16>( in XLA_TEST_F() 131 auto grad_output = ConstantR4FromArray4D<bfloat16>( in XLA_TEST_F()
|
H A D | batch_normalization_test.cc | 220 auto operand = ConstantR4FromArray4D<float>( in XLA_TEST_F() 244 auto operand = ConstantR4FromArray4D<Eigen::half>(&builder, input); in XLA_TEST_F() 273 auto operand = ConstantR4FromArray4D<float>( in XLA_TEST_F() 298 auto operand = ConstantR4FromArray4D<Eigen::half>(&builder, input); in XLA_TEST_F() 392 ConstantR4FromArray4D<float>(&builder, Array4D<float>(2, 2, 2, 1, 0.0f)); in XLA_TEST_F() 400 auto grad_output = ConstantR4FromArray4D<float>( in XLA_TEST_F() 419 auto operand = ConstantR4FromArray4D<Eigen::half>( in XLA_TEST_F() 431 auto grad_output = ConstantR4FromArray4D<Eigen::half>( in XLA_TEST_F()
|
H A D | reverse_test.cc | 129 Rev(ConstantR4FromArray4D<uint8_t>(&b, input), {0, 3}); in XLA_TEST_F() 165 Rev(ConstantR4FromArray4D<float>(&b, input), {0, 1}); in TEST_F()
|
H A D | slice_test.cc | 163 auto original = ConstantR4FromArray4D(&builder, values); in TEST_F() 182 auto original = ConstantR4FromArray4D(&builder, values); in TEST_F() 196 auto original = ConstantR4FromArray4D(&builder, values); in XLA_TEST_F()
|
H A D | binop_scaling_test.cc | 133 auto lhs = ConstantR4FromArray4D(&builder, lhs_array); in TEST_F()
|
H A D | convolution_dimension_numbers_test.cc | 97 auto input = ConstantR4FromArray4D<float>(&builder, *input_array); in XLA_TEST_F()
|
H A D | array_elementwise_ops_test.cc | 1898 auto x = ConstantR4FromArray4D<float>(&builder, values); in XLA_TEST_F() 1909 auto x = ConstantR4FromArray4D<float>(&builder, values); in XLA_TEST_F() 2139 auto array4d = ConstantR4FromArray4D<float>( in XLA_TEST_F() 2155 auto array4d = ConstantR4FromArray4D<float>(&builder, arg); in XLA_TEST_F() 3035 auto a = ConstantR4FromArray4D<float>(&builder, *operand_a_4d); 3036 auto b = ConstantR4FromArray4D<float>(&builder, *operand_b_4d); 3063 auto a = ConstantR4FromArray4D<float>(&builder, *operand_a_4d); in XLA_TEST_F()
|
H A D | constants_test.cc | 156 ConstantR4FromArray4D<float>(&builder, input_array); in TEST_F()
|
H A D | vector_ops_simple_test.cc | 104 auto x = ConstantR4FromArray4D<float>(&builder, exponents); in XLA_TEST_F()
|
H A D | convolution_test.cc | 91 auto lhs = ConstantR4FromArray4D<T>(&builder, *alhs); in RunTest() 92 auto rhs = ConstantR4FromArray4D<T>(&builder, *arhs); in RunTest() 1579 auto filter = ConstantR4FromArray4D(&builder, filter_data); in XLA_TEST_F()
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/lib/ |
H A D | pooling_test.cc | 68 XlaOp input = ConstantR4FromArray4D<float>( in XLA_TEST_F() 81 XlaOp input = ConstantR4FromArray4D<float>( in XLA_TEST_F() 94 XlaOp input = ConstantR4FromArray4D<float>( in XLA_TEST_F() 108 XlaOp input = ConstantR4FromArray4D<float>( in XLA_TEST_F() 124 XlaOp input = ConstantR4FromArray4D<float>( in XLA_TEST_F() 140 XlaOp input = ConstantR4FromArray4D<float>( in XLA_TEST_F() 158 XlaOp input = ConstantR4FromArray4D<float>( in XLA_TEST_F() 173 XlaOp input = ConstantR4FromArray4D<float>( in XLA_TEST_F() 188 XlaOp out_backprop = ConstantR4FromArray4D<float>(&builder, {{{{1.}}}}); in XLA_TEST_F() 206 ConstantR4FromArray4D<float>(&builder, {{{{1., 1.}, {1., 1.}}}}); in XLA_TEST_F() [all …]
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/ |
H A D | xla_builder.h | 1713 XlaOp ConstantR4FromArray4D(XlaBuilder* builder, 2795 XlaOp ConstantR4FromArray4D(XlaBuilder* builder, in ConstantR4FromArray4D() function
|