Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/tests/
Dselect_and_scatter_test.cc262 ConstantR2<int32>(&builder_, {{7, 2, 5, 3, 10, 2}, {3, 8, 9, 3, 4, 2}}); in XLA_TEST_F()
263 const auto source = ConstantR2<int32>(&builder_, {{2, 6}}); in XLA_TEST_F()
274 const auto operand = ConstantR2<float>( in XLA_TEST_F()
276 const auto source = ConstantR2<float>( in XLA_TEST_F()
288 const auto operand = ConstantR2<int32>( in XLA_TEST_F()
292 const auto source = ConstantR2<int32>(&builder_, {{2, 6}}); in XLA_TEST_F()
303 ConstantR2<int32>(&builder_, {{7, 2, 5, 3, 8}, {3, 8, 9, 3, 4}}); in XLA_TEST_F()
304 const auto source = ConstantR2<int32>(&builder_, {{2, 6, 4}}); in XLA_TEST_F()
315 ConstantR2<int32>(&builder_, {{7, 2, 5, 3, 8}, {3, 8, 9, 3, 4}}); in XLA_TEST_F()
316 const auto source = ConstantR2<int32>(&builder_, {{2, 6, 4}}); in XLA_TEST_F()
[all …]
Dclient_test.cc46 Add(ConstantR2<int32>(&b, {{1, 2}, {3, 4}}), in XLA_TEST_F()
47 ConstantR2<int32>(&b, {{10, 20}, {30, 40}})); in XLA_TEST_F()
75 Tuple(&b, {ConstantR2<int32>(&b, {{1, 2}, {3, 4}}), in XLA_TEST_F()
76 ConstantR2<int32>(&b, {{10, 20}, {30, 40}})}); in XLA_TEST_F()
125 ConstantR2<int32>(&b, {{1, 2}, {3, 4}})); in XLA_TEST_F()
Darray_elementwise_ops_test.cc808 auto a = ConstantR2<bool>(&builder, {{false, false}, {true, true}}); in XLA_TEST_F()
809 auto b = ConstantR2<bool>(&builder, {{false, true}, {false, true}}); in XLA_TEST_F()
836 auto a = ConstantR2<int32>(&builder, {{0, -5}, {-1, 5}}); in XLA_TEST_F()
837 auto b = ConstantR2<int32>(&builder, {{1, -6}, {4, 5}}); in XLA_TEST_F()
864 auto a = ConstantR2<uint32>(&builder, {{0, 1}, {3, 8}}); in XLA_TEST_F()
865 auto b = ConstantR2<uint32>(&builder, {{1, 0}, {7, 6}}); in XLA_TEST_F()
892 auto a = ConstantR2<bool>(&builder, {{false, false}, {true, true}}); in XLA_TEST_F()
893 auto b = ConstantR2<bool>(&builder, {{false, true}, {false, true}}); in XLA_TEST_F()
920 auto a = ConstantR2<int32>(&builder, {{0, -1}, {8, 8}}); in XLA_TEST_F()
921 auto b = ConstantR2<int32>(&builder, {{5, -7}, {4, 1}}); in XLA_TEST_F()
[all …]
Dreshape_motion_test.cc47 auto a = ConstantR2<int32>(&builder, {{2, 3, 5}, {7, 11, 13}}); in TEST_F()
48 auto b = ConstantR2<int32>(&builder, {{17, 19}, {23, 29}, {31, 37}}); in TEST_F()
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()
Dcompute_constant_test.cc174 Add(ConstantR2<float>(&b, {{1.0f}}), ConstantR2<float>(&b, {{1.0f}})); in TEST_F()
255 Add(ConstantR2<int32>(&b, {{1, 2}, {3, 4}}), in XLA_TEST_F()
256 ConstantR2<int32>(&b, {{10, 20}, {30, 40}})), in XLA_TEST_F()
Dbroadcast_simple_test.cc190 BroadcastInDim(ConstantR2<float>(&b, {{1.0, 5.0}, {2.0, 6.0}}), {2, 2, 2},
208 BroadcastInDim(ConstantR2<float>(&b, {{1.0, 5.0}, {2.0, 6.0}}), {2, 2, 2},
292 Add(ConstantR2<float>(&b, {{1.0, 5.0}}), in XLA_TEST_F()
735 Add(ConstantR2<float>(&b, {{1.0, 5.0}, {1.0, 5.0}}), in XLA_TEST_F()
750 Add(ConstantR2<float>(&b, {{1.0, 2.0}}), in XLA_TEST_F()
751 ConstantR2<float>(&b, {{1.0, 2.0, 3.0}, {4.0, 5.0, 6.0}})); in XLA_TEST_F()
763 Add(ConstantR2<float>(&b, {{1.0, 2.0}}), in XLA_TEST_F()
764 ConstantR2<float>(&b, {{1.0, 2.0, 3.0}, {4.0, 5.0, 6.0}})); 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()
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()
255 ConstantR2<float>(&builder, constant_matrix)}); in XLA_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.cc195 auto arg = ConstantR2<float>(&builder, {{1.0, -2.0}, {-3.0, 4.0}}); in XLA_TEST_F()
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dquantize_test.cc210 auto x = ConstantR2<uint32>(&builder, {{PackToUint32<uint8>(input[0])[0]}, in XLA_TEST_F()
233 auto x = ConstantR2<uint32>(&builder, {{PackToUint32<uint8>(input[0])[0]}, in XLA_TEST_F()
256 auto x = ConstantR2<uint32>( in XLA_TEST_F()
286 auto x = ConstantR2<uint32>( in XLA_TEST_F()
Darithmetic_test.cc82 XlaOp x = ConstantR2<NativeT>(&builder, input); in TestArgMinMaxImpl()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dresampler_ops.cc55 auto first_term = xla::ConstantR2<float>( in BilinearWeights()
87 auto sign_change = xla::ConstantR2<float>( in BilinearWeights()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.h1588 XlaOp ConstantR2(XlaBuilder* builder,
2580 XlaOp ConstantR2(XlaBuilder* builder, in ConstantR2() function