Home
last modified time | relevance | path

Searched refs:ConstantR4FromArray4D (Results 1 – 14 of 14) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dconvolution_variants_test.cc57 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 Dselect_and_scatter_test.cc365 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 Dbfloat16_test.cc83 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 Dbatch_normalization_test.cc220 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 Dreverse_test.cc129 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 Dslice_test.cc163 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 Dbinop_scaling_test.cc133 auto lhs = ConstantR4FromArray4D(&builder, lhs_array); in TEST_F()
H A Dconvolution_dimension_numbers_test.cc97 auto input = ConstantR4FromArray4D<float>(&builder, *input_array); in XLA_TEST_F()
H A Darray_elementwise_ops_test.cc1898 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 Dconstants_test.cc156 ConstantR4FromArray4D<float>(&builder, input_array); in TEST_F()
H A Dvector_ops_simple_test.cc104 auto x = ConstantR4FromArray4D<float>(&builder, exponents); in XLA_TEST_F()
H A Dconvolution_test.cc91 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 Dpooling_test.cc68 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 Dxla_builder.h1713 XlaOp ConstantR4FromArray4D(XlaBuilder* builder,
2795 XlaOp ConstantR4FromArray4D(XlaBuilder* builder, in ConstantR4FromArray4D() function