Home
last modified time | relevance | path

Searched refs:ConstantR2FromArray2D (Results 1 – 13 of 13) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dconcat_test.cc146 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 …]
H A Ddot_operation_test.cc114 auto lhs = ConstantR2FromArray2D<T>(&builder, {{3.0f, 4.0f}}); in XLA_TYPED_TEST()
151 auto lhs = ConstantR2FromArray2D<T>(&builder, Array2D<T>(0, 2)); in XLA_TYPED_TEST()
152 auto rhs = ConstantR2FromArray2D<T>(&builder, Array2D<T>(2, 0)); in XLA_TYPED_TEST()
162 auto lhs = ConstantR2FromArray2D<T>(&builder, Array2D<T>(0, 2)); in XLA_TYPED_TEST()
163 auto rhs = ConstantR2FromArray2D<T>( in XLA_TYPED_TEST()
174 auto lhs = ConstantR2FromArray2D<T>( in XLA_TYPED_TEST()
176 auto rhs = ConstantR2FromArray2D<T>(&builder, Array2D<T>(2, 0)); in XLA_TYPED_TEST()
186 auto lhs = ConstantR2FromArray2D<T>(&builder, Array2D<T>(2, 0)); in XLA_TYPED_TEST()
187 auto rhs = ConstantR2FromArray2D<T>(&builder, Array2D<T>(0, 2)); in XLA_TYPED_TEST()
574 ConstantR2FromArray2D<T>(&builder, {{1.0f, 2.0f}, {3.0f, 4.0f}}); in XLA_TYPED_TEST()
[all …]
H A Dbinop_scaling_test.cc36 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()
H A Dtranspose_test.cc41 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()
H A Dmatrix_ops_simple_test.cc58 auto data = ConstantR2FromArray2D<T>(&builder, { in XLA_TYPED_TEST()
87 auto data = ConstantR2FromArray2D<T>(&builder, { in XLA_TYPED_TEST()
102 auto lhs = ConstantR2FromArray2D<T>(&builder, { in XLA_TYPED_TEST()
106 auto rhs = ConstantR2FromArray2D<T>(&builder, { in XLA_TYPED_TEST()
138 auto lhs = ConstantR2FromArray2D<T>(&builder, *alhs); in TestImpl()
139 auto rhs = ConstantR2FromArray2D<T>(&builder, *arhs); in TestImpl()
H A Dslice_test.cc81 auto original = ConstantR2FromArray2D<float>(&builder, Array2D<float>(0, 0)); in XLA_TEST_F()
89 auto original = ConstantR2FromArray2D<float>(&builder, Array2D<float>(0, 20)); in XLA_TEST_F()
97 auto original = ConstantR2FromArray2D<float>(&builder, Array2D<float>(3, 0)); in XLA_TEST_F()
112 auto original = ConstantR2FromArray2D<float>(&builder, values); in XLA_TEST_F()
130 auto original = ConstantR2FromArray2D<float>(&builder, values); in TEST_F()
151 auto original = ConstantR2FromArray2D<float>(&builder, values); in TEST_F()
172 auto original = ConstantR2FromArray2D(&builder, values); in TEST_F()
H A Dconstants_test.cc99 ConstantR2FromArray2D<float>(&builder, Array2D<float>(0, 2)); in TEST_F()
109 ConstantR2FromArray2D<float>(&builder, *constant); in TEST_F()
H A Dtuple_test.cc140 ConstantR2FromArray2D<float>(&builder, Array2D<float>(0, 101))}); in XLA_TEST_F()
H A Dreduce_test.cc972 auto m = ConstantR2FromArray2D(&builder, initializer); in XLA_TEST_F()
986 auto m = ConstantR2FromArray2D(&builder, initializer); in XLA_TEST_F()
H A Darray_elementwise_ops_test.cc2064 auto v = ConstantR2FromArray2D<float>(&builder, Array2D<float>(0, 2)); in XLA_TEST_F()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/lib/
H A Dquantize_test.cc322 auto x = ConstantR2FromArray2D<uint32_t>(&builder, input_packed); in XLA_TEST_F()
336 auto x = ConstantR2FromArray2D<uint32_t>(&builder, input_packed); in XLA_TEST_F()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dimage_ops.cc392 row_iou, xla::ConstantR2FromArray2D<bool>(builder, {{false}}), in operator ()()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/
H A Dxla_builder.h1699 XlaOp ConstantR2FromArray2D(XlaBuilder* builder,
2766 XlaOp ConstantR2FromArray2D(XlaBuilder* builder, in ConstantR2FromArray2D() function