| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | convolution_variants_test.cc | 58 auto input = ConstantR4FromArray4D<float>(&builder, input_array); in XLA_TEST_F()
61 auto filter = ConstantR4FromArray4D<float>(&builder, filter_array); in XLA_TEST_F()
73 auto input = ConstantR4FromArray4D<float>(&builder, input_array); in XLA_TEST_F()
76 auto filter = ConstantR4FromArray4D<float>(&builder, filter_array); in XLA_TEST_F()
89 auto input = ConstantR4FromArray4D<float>(&builder, input_array); in XLA_TEST_F()
92 auto filter = ConstantR4FromArray4D<float>(&builder, filter_array); in XLA_TEST_F()
105 auto input = ConstantR4FromArray4D<float>(&builder, input_array); in XLA_TEST_F()
108 auto filter = ConstantR4FromArray4D<float>(&builder, filter_array); in XLA_TEST_F()
120 auto input = ConstantR4FromArray4D<float>(&builder, input_array); in XLA_TEST_F()
123 auto filter = ConstantR4FromArray4D<float>(&builder, filter_array); in XLA_TEST_F()
[all …]
|
| D | select_and_scatter_test.cc | 363 auto operand = ConstantR4FromArray4D(&builder_, o); in TEST_F()
368 auto source = ConstantR4FromArray4D(&builder_, s); in TEST_F()
388 auto operand = ConstantR4FromArray4D(&builder_, o); in TEST_F()
393 auto source = ConstantR4FromArray4D(&builder_, s); in TEST_F()
413 auto operand = ConstantR4FromArray4D(&builder_, o); in TEST_F()
418 auto source = ConstantR4FromArray4D(&builder_, s); in TEST_F()
435 auto operand = ConstantR4FromArray4D(&builder_, o); in TEST_F()
439 auto source = ConstantR4FromArray4D(&builder_, s); in TEST_F()
|
| D | bfloat16_test.cc | 85 auto operand = ConstantR4FromArray4D<bfloat16>( in XLA_TEST_F()
121 auto operand = ConstantR4FromArray4D<bfloat16>( in XLA_TEST_F()
133 auto grad_output = ConstantR4FromArray4D<bfloat16>( in XLA_TEST_F()
|
| D | reverse_test.cc | 130 Rev(ConstantR4FromArray4D<uint8>(&b, input), {0, 3}); in XLA_TEST_F()
166 Rev(ConstantR4FromArray4D<float>(&b, input), {0, 1}); in TEST_F()
|
| D | batch_normalization_test.cc | 236 auto operand = ConstantR4FromArray4D<float>( in XLA_TEST_P()
260 auto operand = ConstantR4FromArray4D<Eigen::half>(&builder, input); in XLA_TEST_P()
289 auto operand = ConstantR4FromArray4D<float>( in XLA_TEST_P()
314 auto operand = ConstantR4FromArray4D<Eigen::half>(&builder, input); in XLA_TEST_P()
408 ConstantR4FromArray4D<float>(&builder, Array4D<float>(2, 2, 2, 1, 0.0f)); in XLA_TEST_P()
416 auto grad_output = ConstantR4FromArray4D<float>( in XLA_TEST_P()
435 auto operand = ConstantR4FromArray4D<Eigen::half>( in XLA_TEST_P()
447 auto grad_output = ConstantR4FromArray4D<Eigen::half>( in XLA_TEST_P()
|
| D | slice_test.cc | 164 auto original = ConstantR4FromArray4D(&builder, values); in TEST_F()
183 auto original = ConstantR4FromArray4D(&builder, values); in TEST_F()
197 auto original = ConstantR4FromArray4D(&builder, values); in XLA_TEST_F()
|
| D | binop_scaling_test.cc | 133 auto lhs = ConstantR4FromArray4D(&builder, lhs_array); in TEST_F()
|
| D | convolution_dimension_numbers_test.cc | 100 auto input = ConstantR4FromArray4D<float>(&builder, *input_array); in XLA_TEST_F()
|
| D | constants_test.cc | 156 ConstantR4FromArray4D<float>(&builder, input_array); in TEST_F()
|
| D | array_elementwise_ops_test.cc | 1880 auto x = ConstantR4FromArray4D<float>(&builder, values); in XLA_TEST_F()
1891 auto x = ConstantR4FromArray4D<float>(&builder, values); in XLA_TEST_F()
2110 auto array4d = ConstantR4FromArray4D<float>( in XLA_TEST_F()
2126 auto array4d = ConstantR4FromArray4D<float>(&builder, arg); in XLA_TEST_F()
2956 auto a = ConstantR4FromArray4D<float>(&builder, *operand_a_4d);
2957 auto b = ConstantR4FromArray4D<float>(&builder, *operand_b_4d);
2984 auto a = ConstantR4FromArray4D<float>(&builder, *operand_a_4d); in XLA_TEST_F()
|
| D | vector_ops_simple_test.cc | 101 auto x = ConstantR4FromArray4D<float>(&builder, exponents); in XLA_TEST_F()
|
| D | convolution_test.cc | 92 auto lhs = ConstantR4FromArray4D<T>(&builder, *alhs); in RunTest()
93 auto rhs = ConstantR4FromArray4D<T>(&builder, *arhs); in RunTest()
1605 auto filter = ConstantR4FromArray4D(&builder, filter_data); in XLA_TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| 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 …]
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 1615 XlaOp ConstantR4FromArray4D(XlaBuilder* builder,
2637 XlaOp ConstantR4FromArray4D(XlaBuilder* builder, in ConstantR4FromArray4D() function
|