| /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 | 143 auto a = ConstantR2FromArray2D(&builder, Array2D<float>(0, 0)); in XLA_TEST_F()
144 auto b = ConstantR2FromArray2D(&builder, Array2D<float>(0, 0)); in XLA_TEST_F()
156 auto a = ConstantR2FromArray2D(&builder, *a_array); in XLA_TEST_F()
157 auto b = ConstantR2FromArray2D(&builder, *b_array); in XLA_TEST_F()
171 auto a = ConstantR2FromArray2D(&builder, *a_array); in XLA_TEST_F()
172 auto b = ConstantR2FromArray2D(&builder, *b_array); in XLA_TEST_F()
184 auto a = ConstantR2FromArray2D(&builder, Array2D<float>(2, 0)); in XLA_TEST_F()
185 auto b = ConstantR2FromArray2D(&builder, *b_array); in XLA_TEST_F()
195 auto a = ConstantR2FromArray2D(&builder, *a_array); in XLA_TEST_F()
196 auto b = ConstantR2FromArray2D(&builder, *b_array); in XLA_TEST_F()
[all …]
|
| D | dot_operation_test.cc | 108 auto lhs = ConstantR2FromArray2D<T>(&builder, {{3.0f, 4.0f}}); in XLA_TYPED_TEST()
145 auto lhs = ConstantR2FromArray2D<T>(&builder, Array2D<T>(0, 2)); in XLA_TYPED_TEST()
146 auto rhs = ConstantR2FromArray2D<T>(&builder, Array2D<T>(2, 0)); in XLA_TYPED_TEST()
156 auto lhs = ConstantR2FromArray2D<T>(&builder, Array2D<T>(0, 2)); in XLA_TYPED_TEST()
157 auto rhs = ConstantR2FromArray2D<T>( in XLA_TYPED_TEST()
168 auto lhs = ConstantR2FromArray2D<T>( in XLA_TYPED_TEST()
170 auto rhs = ConstantR2FromArray2D<T>(&builder, Array2D<T>(2, 0)); in XLA_TYPED_TEST()
180 auto lhs = ConstantR2FromArray2D<T>(&builder, Array2D<T>(2, 0)); in XLA_TYPED_TEST()
181 auto rhs = ConstantR2FromArray2D<T>(&builder, Array2D<T>(0, 2)); in XLA_TYPED_TEST()
531 ConstantR2FromArray2D<T>(&builder, {{1.0f, 2.0f}, {3.0f, 4.0f}}); in XLA_TYPED_TEST()
[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 | 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 | reduce_test.cc | 964 auto m = ConstantR2FromArray2D(&builder, initializer); in XLA_TEST_F()
977 auto m = ConstantR2FromArray2D(&builder, initializer); in XLA_TEST_F()
|
| D | tuple_test.cc | 140 ConstantR2FromArray2D<float>(&builder, Array2D<float>(0, 101))}); in XLA_TEST_F()
|
| D | array_elementwise_ops_test.cc | 1929 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 | 383 row_iou, xla::ConstantR2FromArray2D<bool>(builder, {{false}}), in operator ()()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 1087 XlaOp ConstantR2FromArray2D(XlaBuilder* builder,
1976 XlaOp ConstantR2FromArray2D(XlaBuilder* builder, in ConstantR2FromArray2D() function
|