/external/tensorflow/tensorflow/compiler/xla/tests/ |
D | select_and_scatter_test.cc | 262 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 …]
|
D | client_test.cc | 46 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()
|
D | array_elementwise_ops_test.cc | 808 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 …]
|
D | reshape_motion_test.cc | 47 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()
|
D | pred_test.cc | 88 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()
|
D | compute_constant_test.cc | 174 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()
|
D | broadcast_simple_test.cc | 190 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()
|
D | vector_ops_reduce_test.cc | 107 auto x = ConstantR2<float>(&builder_, { in TEST_F() 123 auto x = ConstantR2<float>(&builder_, { in TEST_F()
|
D | tuple_test.cc | 89 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()
|
D | multidimensional_slice_test.cc | 36 auto original = ConstantR2<float>( in XLA_TEST_F()
|
D | binop_scaling_test.cc | 98 auto rhs = ConstantR2<float>(&builder, { in TEST_F()
|
D | transpose_test.cc | 65 auto lhs = ConstantR2<float>(&builder, { in TEST_F()
|
D | unary_op_test.cc | 195 auto arg = ConstantR2<float>(&builder, {{1.0, -2.0}, {-3.0, 4.0}}); in XLA_TEST_F()
|
/external/tensorflow/tensorflow/compiler/xla/client/lib/ |
D | quantize_test.cc | 210 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()
|
D | arithmetic_test.cc | 82 XlaOp x = ConstantR2<NativeT>(&builder, input); in TestArgMinMaxImpl()
|
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
D | resampler_ops.cc | 55 auto first_term = xla::ConstantR2<float>( in BilinearWeights() 87 auto sign_change = xla::ConstantR2<float>( in BilinearWeights()
|
/external/tensorflow/tensorflow/compiler/xla/client/ |
D | xla_builder.h | 1588 XlaOp ConstantR2(XlaBuilder* builder, 2580 XlaOp ConstantR2(XlaBuilder* builder, in ConstantR2() function
|