| /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 | 352 auto operand = ConstantR4FromArray4D(&builder_, o); in TEST_F()
357 auto source = ConstantR4FromArray4D(&builder_, s); in TEST_F()
377 auto operand = ConstantR4FromArray4D(&builder_, o); in TEST_F()
382 auto source = ConstantR4FromArray4D(&builder_, s); in TEST_F()
402 auto operand = ConstantR4FromArray4D(&builder_, o); in TEST_F()
407 auto source = ConstantR4FromArray4D(&builder_, s); in TEST_F()
424 auto operand = ConstantR4FromArray4D(&builder_, o); in TEST_F()
428 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 | 129 Rev(ConstantR4FromArray4D<uint8>(&b, input), {0, 3}); in XLA_TEST_F()
165 Rev(ConstantR4FromArray4D<float>(&b, input), {0, 1}); in TEST_F()
|
| 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 | batch_normalization_test.cc | 235 auto operand = ConstantR4FromArray4D<float>( in XLA_TEST_P()
258 auto operand = ConstantR4FromArray4D<float>( in XLA_TEST_P()
347 ConstantR4FromArray4D<float>(&builder, Array4D<float>(2, 2, 2, 1, 0.0f)); in XLA_TEST_P()
355 auto grad_output = ConstantR4FromArray4D<float>( in XLA_TEST_P()
|
| 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 | 1774 auto x = ConstantR4FromArray4D<float>(&builder, values); in XLA_TEST_F()
1785 auto x = ConstantR4FromArray4D<float>(&builder, values); in XLA_TEST_F()
2004 auto array4d = ConstantR4FromArray4D<float>( in XLA_TEST_F()
2020 auto array4d = ConstantR4FromArray4D<float>(&builder, arg); in XLA_TEST_F()
2850 auto a = ConstantR4FromArray4D<float>(&builder, *operand_a_4d);
2851 auto b = ConstantR4FromArray4D<float>(&builder, *operand_b_4d);
2878 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()
1919 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 | 1101 XlaOp ConstantR4FromArray4D(XlaBuilder* builder,
2005 XlaOp ConstantR4FromArray4D(XlaBuilder* builder, in ConstantR4FromArray4D() function
|