| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| 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()
|
| 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 …]
|
| 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()
|
| D | dot_operation_test.cc | 115 auto lhs = ConstantR2FromArray2D<T>(&builder, {{3.0f, 4.0f}}); in XLA_TYPED_TEST()
152 auto lhs = ConstantR2FromArray2D<T>(&builder, Array2D<T>(0, 2)); in XLA_TYPED_TEST()
153 auto rhs = ConstantR2FromArray2D<T>(&builder, Array2D<T>(2, 0)); in XLA_TYPED_TEST()
163 auto lhs = ConstantR2FromArray2D<T>(&builder, Array2D<T>(0, 2)); in XLA_TYPED_TEST()
164 auto rhs = ConstantR2FromArray2D<T>( in XLA_TYPED_TEST()
175 auto lhs = ConstantR2FromArray2D<T>( in XLA_TYPED_TEST()
177 auto rhs = ConstantR2FromArray2D<T>(&builder, Array2D<T>(2, 0)); in XLA_TYPED_TEST()
187 auto lhs = ConstantR2FromArray2D<T>(&builder, Array2D<T>(2, 0)); in XLA_TYPED_TEST()
188 auto rhs = ConstantR2FromArray2D<T>(&builder, Array2D<T>(0, 2)); in XLA_TYPED_TEST()
575 ConstantR2FromArray2D<T>(&builder, {{1.0f, 2.0f}, {3.0f, 4.0f}}); in XLA_TYPED_TEST()
[all …]
|
| D | matrix_ops_simple_test.cc | 60 auto data = ConstantR2FromArray2D<T>(&builder, { in XLA_TYPED_TEST()
89 auto data = ConstantR2FromArray2D<T>(&builder, { in XLA_TYPED_TEST()
104 auto lhs = ConstantR2FromArray2D<T>(&builder, { in XLA_TYPED_TEST()
108 auto rhs = ConstantR2FromArray2D<T>(&builder, { in XLA_TYPED_TEST()
140 auto lhs = ConstantR2FromArray2D<T>(&builder, *alhs); in TestImpl()
141 auto rhs = ConstantR2FromArray2D<T>(&builder, *arhs); in TestImpl()
|
| D | slice_test.cc | 82 auto original = ConstantR2FromArray2D<float>(&builder, Array2D<float>(0, 0)); in XLA_TEST_F()
90 auto original = ConstantR2FromArray2D<float>(&builder, Array2D<float>(0, 20)); in XLA_TEST_F()
98 auto original = ConstantR2FromArray2D<float>(&builder, Array2D<float>(3, 0)); in XLA_TEST_F()
113 auto original = ConstantR2FromArray2D<float>(&builder, values); in XLA_TEST_F()
131 auto original = ConstantR2FromArray2D<float>(&builder, values); in TEST_F()
152 auto original = ConstantR2FromArray2D<float>(&builder, values); in TEST_F()
173 auto original = ConstantR2FromArray2D(&builder, values); in TEST_F()
|
| D | constants_test.cc | 99 ConstantR2FromArray2D<float>(&builder, Array2D<float>(0, 2)); in TEST_F()
109 ConstantR2FromArray2D<float>(&builder, *constant); in TEST_F()
|
| D | tuple_test.cc | 140 ConstantR2FromArray2D<float>(&builder, Array2D<float>(0, 101))}); in XLA_TEST_F()
|
| D | reduce_test.cc | 965 auto m = ConstantR2FromArray2D(&builder, initializer); in XLA_TEST_F()
978 auto m = ConstantR2FromArray2D(&builder, initializer); in XLA_TEST_F()
|
| D | array_elementwise_ops_test.cc | 2035 auto v = ConstantR2FromArray2D<float>(&builder, Array2D<float>(0, 2)); in XLA_TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | quantize_test.cc | 313 auto x = ConstantR2FromArray2D<uint32>(&builder, input_packed); in XLA_TEST_F()
327 auto x = ConstantR2FromArray2D<uint32>(&builder, input_packed); in XLA_TEST_F()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | image_ops.cc | 385 row_iou, xla::ConstantR2FromArray2D<bool>(builder, {{false}}), in operator ()()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 1601 XlaOp ConstantR2FromArray2D(XlaBuilder* builder,
2608 XlaOp ConstantR2FromArray2D(XlaBuilder* builder, in ConstantR2FromArray2D() function
|