Home
last modified time | relevance | path

Searched refs:ConstantR2 (Results 1 – 18 of 18) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dselect_and_scatter_test.cc264 ConstantR2<int32_t>(&builder_, {{7, 2, 5, 3, 10, 2}, {3, 8, 9, 3, 4, 2}}); in XLA_TEST_F()
265 const auto source = ConstantR2<int32_t>(&builder_, {{2, 6}}); in XLA_TEST_F()
276 const auto operand = ConstantR2<float>( in XLA_TEST_F()
278 const auto source = ConstantR2<float>( in XLA_TEST_F()
290 const auto operand = ConstantR2<int32_t>( in XLA_TEST_F()
294 const auto source = ConstantR2<int32_t>(&builder_, {{2, 6}}); in XLA_TEST_F()
305 ConstantR2<int32_t>(&builder_, {{7, 2, 5, 3, 8}, {3, 8, 9, 3, 4}}); in XLA_TEST_F()
306 const auto source = ConstantR2<int32_t>(&builder_, {{2, 6, 4}}); in XLA_TEST_F()
317 ConstantR2<int32_t>(&builder_, {{7, 2, 5, 3, 8}, {3, 8, 9, 3, 4}}); in XLA_TEST_F()
318 const auto source = ConstantR2<int32_t>(&builder_, {{2, 6, 4}}); in XLA_TEST_F()
[all …]
Dclient_test.cc45 Add(ConstantR2<int32_t>(&b, {{1, 2}, {3, 4}}), in XLA_TEST_F()
46 ConstantR2<int32_t>(&b, {{10, 20}, {30, 40}})); in XLA_TEST_F()
74 Tuple(&b, {ConstantR2<int32_t>(&b, {{1, 2}, {3, 4}}), in XLA_TEST_F()
75 ConstantR2<int32_t>(&b, {{10, 20}, {30, 40}})}); in XLA_TEST_F()
125 ConstantR2<int32_t>(&b, {{1, 2}, {3, 4}})); in XLA_TEST_F()
Darray_elementwise_ops_test.cc815 auto a = ConstantR2<bool>(&builder, {{false, false}, {true, true}}); in XLA_TEST_F()
816 auto b = ConstantR2<bool>(&builder, {{false, true}, {false, true}}); in XLA_TEST_F()
843 auto a = ConstantR2<int32_t>(&builder, {{0, -5}, {-1, 5}}); in XLA_TEST_F()
844 auto b = ConstantR2<int32_t>(&builder, {{1, -6}, {4, 5}}); in XLA_TEST_F()
871 auto a = ConstantR2<uint32_t>(&builder, {{0, 1}, {3, 8}}); in XLA_TEST_F()
872 auto b = ConstantR2<uint32_t>(&builder, {{1, 0}, {7, 6}}); in XLA_TEST_F()
899 auto a = ConstantR2<bool>(&builder, {{false, false}, {true, true}}); in XLA_TEST_F()
900 auto b = ConstantR2<bool>(&builder, {{false, true}, {false, true}}); in XLA_TEST_F()
927 auto a = ConstantR2<int32_t>(&builder, {{0, -1}, {8, 8}}); in XLA_TEST_F()
928 auto b = ConstantR2<int32_t>(&builder, {{5, -7}, {4, 1}}); in XLA_TEST_F()
[all …]
Dpred_test.cc88 ConstantR2<bool>(&builder, {{false, true, true}, {true, false, false}}); in TEST_F()
119 auto a = ConstantR2<bool>(&builder, { in TEST_F()
130 auto a = ConstantR2<bool>(&builder, { in TEST_F()
Dreshape_motion_test.cc46 auto a = ConstantR2<int32_t>(&builder, {{2, 3, 5}, {7, 11, 13}}); in TEST_F()
47 auto b = ConstantR2<int32_t>(&builder, {{17, 19}, {23, 29}, {31, 37}}); in TEST_F()
Dtuple_test.cc89 ConstantR2<float>(&builder, constant_matrix)}); in XLA_TEST_F()
128 ConstantR2<float>(&builder, constant_matrix)}); in XLA_TEST_F()
167 ConstantR2<float>(&builder, constant_matrix)}); in XLA_TEST_F()
196 ConstantR2<float>(&builder, constant_matrix)}); in XLA_TEST_F()
231 ConstantR2<float>(&builder, constant_matrix)}); in XLA_TEST_F()
Dbroadcast_simple_test.cc188 BroadcastInDim(ConstantR2<float>(&b, {{1.0, 5.0}, {2.0, 6.0}}), {2, 2, 2},
206 BroadcastInDim(ConstantR2<float>(&b, {{1.0, 5.0}, {2.0, 6.0}}), {2, 2, 2},
290 Add(ConstantR2<float>(&b, {{1.0, 5.0}}), in XLA_TEST_F()
733 Add(ConstantR2<float>(&b, {{1.0, 5.0}, {1.0, 5.0}}), in XLA_TEST_F()
748 Add(ConstantR2<float>(&b, {{1.0, 2.0}}), in XLA_TEST_F()
749 ConstantR2<float>(&b, {{1.0, 2.0, 3.0}, {4.0, 5.0, 6.0}})); in XLA_TEST_F()
761 Add(ConstantR2<float>(&b, {{1.0, 2.0}}), in XLA_TEST_F()
762 ConstantR2<float>(&b, {{1.0, 2.0, 3.0}, {4.0, 5.0, 6.0}})); in XLA_TEST_F()
Dcompute_constant_test.cc173 Add(ConstantR2<float>(&b, {{1.0f}}), ConstantR2<float>(&b, {{1.0f}})); in TEST_F()
255 Add(ConstantR2<int32_t>(&b, {{1, 2}, {3, 4}}), in XLA_TEST_F()
256 ConstantR2<int32_t>(&b, {{10, 20}, {30, 40}})), in XLA_TEST_F()
Dvector_ops_reduce_test.cc107 auto x = ConstantR2<float>(&builder_, { in TEST_F()
123 auto x = ConstantR2<float>(&builder_, { in TEST_F()
Dmultidimensional_slice_test.cc36 auto original = ConstantR2<float>( in XLA_TEST_F()
Dbinop_scaling_test.cc98 auto rhs = ConstantR2<float>(&builder, { in TEST_F()
Dtranspose_test.cc65 auto lhs = ConstantR2<float>(&builder, { in TEST_F()
Dunary_op_test.cc194 auto arg = ConstantR2<float>(&builder, {{1.0, -2.0}, {-3.0, 4.0}}); in XLA_TEST_F()
Dvalue_inference_test.cc143 auto c = ConstantR2<int32_t>(&b, {{0, 0}}); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dquantize_test.cc216 ConstantR2<uint32_t>(&builder, {{PackToUint32<uint8_t>(input[0])[0]}, in XLA_TEST_F()
240 ConstantR2<uint32_t>(&builder, {{PackToUint32<uint8_t>(input[0])[0]}, in XLA_TEST_F()
263 auto x = ConstantR2<uint32_t>( in XLA_TEST_F()
294 auto x = ConstantR2<uint32_t>( in XLA_TEST_F()
Darithmetic_test.cc53 XlaOp x = ConstantR2<NativeT>(&builder, input); in TestArgMinMax()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dresampler_ops.cc57 auto first_term = xla::ConstantR2<float>( in BilinearWeights()
89 auto sign_change = xla::ConstantR2<float>( in BilinearWeights()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.h1686 XlaOp ConstantR2(XlaBuilder* builder,
2738 XlaOp ConstantR2(XlaBuilder* builder, in ConstantR2() function