Home
last modified time | relevance | path

Searched refs:ConstantR1 (Results 1 – 25 of 46) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/tests/
Darray_elementwise_ops_test.cc54 auto a = ConstantR1<float>(&builder, {}); in XLA_TEST_F()
62 auto a = ConstantR1<float>(&builder, {-2.5f, 3.14f, 2.25f, -10.0f, 6.0f}); in XLA_TEST_F()
71 auto a = ConstantR1<int32>(&builder, in XLA_TEST_F()
87 auto a = ConstantR1<complex64>(&builder, {}); in XLA_TEST_F()
95 auto a = ConstantR1<complex64>( in XLA_TEST_F()
107 ConstantR1<int64>(&builder, { in XLA_TEST_F()
134 auto a = ConstantR1<float>(&builder, {}); in XLA_TEST_F()
167 auto a = ConstantR1<float>(&builder, in XLA_TEST_F()
177 auto a = ConstantR1<float>(&builder, {-2.5f, 3.14f, 2.25f, -10.0f, 6.0f}); in XLA_TEST_F()
178 auto b = ConstantR1<float>(&builder, {100.0f, 3.13f, 2.75f, 10.5f, -999.0f}); in XLA_TEST_F()
[all …]
Dselect_test.cc68 auto pred = ConstantR1<bool>(&builder, {}); in XLA_TEST_F()
69 auto on_true = ConstantR1<float>(&builder, {}); in XLA_TEST_F()
70 auto on_false = ConstantR1<float>(&builder, {}); in XLA_TEST_F()
78 auto pred = ConstantR1<bool>(&builder, {false, true, false, true, false}); in TEST_F()
80 ConstantR1<float>(&builder, {-2.5f, 25.5f, 2.25f, -10.0f, 6.0f}); in TEST_F()
82 ConstantR1<float>(&builder, {10.0f, 5.0f, 1.0f, 10.0f, -6.0f}); in TEST_F()
93 auto v1 = ConstantR1<int32>(&builder, {}); in XLA_TEST_F()
94 auto v2 = ConstantR1<int32>(&builder, {}); in XLA_TEST_F()
96 auto on_true = ConstantR1<float>(&builder, {}); in XLA_TEST_F()
97 auto on_false = ConstantR1<float>(&builder, {}); in XLA_TEST_F()
[all …]
Dvector_ops_simple_test.cc53 auto x = ConstantR1<float>( in XLA_TEST_F()
72 auto x = ConstantR1<float>(&builder, exponents); in XLA_TEST_F()
110 auto x = ConstantR1<float>( in XLA_TEST_F()
121 auto x = ConstantR1<int32>(&builder, {2, -2, 12, -4, 5, 20, -15, 0, -2, 1}); in XLA_TEST_F()
130 auto x = ConstantR1<uint32>( in XLA_TEST_F()
140 auto x = ConstantR1<float>(&builder, in XLA_TEST_F()
154 auto x = ConstantR1<float>( in XLA_TEST_F()
156 auto y = ConstantR1<float>( in XLA_TEST_F()
167 auto x = ConstantR1<float>( in XLA_TEST_F()
169 auto y = ConstantR1<float>( in XLA_TEST_F()
[all …]
Dtuple_test.cc88 ConstantR1<float>(&builder, constant_vector), in XLA_TEST_F()
103 {ConstantR0<float>(&builder, 7.0), ConstantR1<float>(&builder, {})}); in XLA_TEST_F()
127 Tuple(&builder, {ConstantR1<float>(&builder, constant_vector), in XLA_TEST_F()
139 {ConstantR1<float>(&builder, {}), in XLA_TEST_F()
147 auto value = ConstantR1<float>(&builder, {4.5f}); in XLA_TEST_F()
166 Tuple(&builder, {ConstantR1<float>(&builder, constant_vector), in XLA_TEST_F()
195 Tuple(&builder, {ConstantR1<float>(&builder, constant_vector), in XLA_TEST_F()
254 Tuple(&builder, {ConstantR1<float>(&builder, constant_vector), in XLA_TEST_F()
284 auto tuple12 = Tuple(&builder, {ConstantR1<float>(&builder, vec1), in XLA_TEST_F()
285 ConstantR1<float>(&builder, vec2)}); in XLA_TEST_F()
[all …]
Dconvert_test.cc50 auto a = ConstantR1<int32>(&builder, {42, 64}); in TEST_F()
59 auto a = ConstantR1<int32>(&builder, {42, 64}); in TEST_F()
68 auto a = ConstantR1<int32>(&builder, {42, 0, -64}); in TEST_F()
77 auto a = ConstantR1<uint32>(&builder, {42, 64}); in TEST_F()
86 auto a = ConstantR1<uint32>(&builder, {42, 64}); in TEST_F()
95 auto a = ConstantR1<uint32>(&builder, {42, 0, 64}); in TEST_F()
104 auto a = ConstantR1<float>(&builder, {42.0f, 64.0f}); in TEST_F()
113 auto a = ConstantR1<float>(&builder, {42.0f, 0.0f, 64.0f}); in TEST_F()
122 auto a = ConstantR1<int32>(&builder, {42, 64}); in TEST_F()
131 auto a = ConstantR1<bool>(&builder, {true, false, true}); in TEST_F()
[all …]
Dconcat_test.cc52 auto a = ConstantR1<float>(&builder, {42.0, 64.0}); in XLA_TEST_F()
61 auto a = ConstantR1<float>(&builder, {}); in XLA_TEST_F()
83 auto a = ConstantR1<float>(&builder, {}); in XLA_TEST_F()
84 auto b = ConstantR1<float>(&builder, {}); in XLA_TEST_F()
93 auto a = ConstantR1<float>(&builder, {}); in XLA_TEST_F()
94 auto b = ConstantR1<float>(&builder, {256.0}); in XLA_TEST_F()
103 auto a = ConstantR1<float>(&builder, {42.0, 64.0}); in XLA_TEST_F()
104 auto b = ConstantR1<float>(&builder, {}); in XLA_TEST_F()
113 auto a = ConstantR1<float>(&builder, {42.0, 64.0}); in XLA_TEST_F()
114 auto b = ConstantR1<float>(&builder, {256.0}); in XLA_TEST_F()
[all …]
Dbitcast_convert_test.cc46 auto a = ConstantR1<int32>(&builder, {42, 64}); in TEST_F()
55 auto a = ConstantR1<float>(&builder, {42.0f, 64.0f}); in TEST_F()
65 ConstantR1<int32>(&builder, {0, static_cast<int32>(0x80000000), in TEST_F()
76 auto a = ConstantR1<int32>(&builder, {}); in XLA_TEST_F()
85 auto a = ConstantR1<float>(&builder, {42.6, 64.4}); in TEST_F()
94 auto a = ConstantR1<int32>(&builder, {std::numeric_limits<int32>::min(), in TEST_F()
107 auto a = ConstantR1<float>(&builder, {42.0f, 64.0f}); in TEST_F()
119 auto a = ConstantR1<int32>(&builder, {0x42280000, 0x42800000}); in TEST_F()
133 auto input = ConstantR1<int32>(&builder, {0x42280000}); in TEST_F()
Dunary_op_test.cc41 auto arg = ConstantR1<T>(&builder, {}); in AbsSize0TestHelper()
54 auto arg = ConstantR1<T>(&builder, {-2, 25, 0, -123, inf<T>(), -inf<T>()}); in AbsTestHelper()
63 auto arg = ConstantR1<T>( in SignTestHelper()
75 auto arg = ConstantR1<T>(&builder, {-2, 25, 0, -123}); in SignAbsTestHelper()
97 auto arg = ConstantR1<complex64>(&builder, {{-2, 0}, in AbsTestHelper()
113 auto arg = ConstantR1<complex64>( in SignTestHelper()
127 ConstantR1<complex64>(&builder, {{-2, 0}, {0, 25}, {0, 0}, {-0.4, 0.3}}); in SignAbsTestHelper()
205 auto lhs = ConstantR1<int32>(&builder, {0, 1}); in XLA_TEST_F()
206 auto rhs = ConstantR1<int32>(&builder, {1, 1}); in XLA_TEST_F()
214 auto lhs = ConstantR1<int32>(&builder, {0, 1}); in XLA_TEST_F()
[all …]
Ddeconstruct_tuple_test.cc57 auto const1 = ConstantR1<float>(&builder, {1.0, 2.0, 3.0, 4.0}); in TEST_F()
58 auto const2 = ConstantR1<float>(&builder, {2.0, 4.0, 6.0, 8.0}); in TEST_F()
76 auto const1 = ConstantR1<float>(&builder, {1.0, 2.0, 3.0, 4.0}); in TEST_F()
77 auto const2 = ConstantR1<float>(&builder, {2.0, 4.0, 6.0, 8.0}); in TEST_F()
106 auto const1 = ConstantR1<float>(&builder, {1.0, 2.0, 3.0, 4.0}); in XLA_TEST_F()
107 auto const2 = ConstantR1<float>(&builder, {2.0, 4.0, 6.0, 8.0}); in XLA_TEST_F()
132 auto const1 = ConstantR1<float>(&builder, {1.0, 2.0, 3.0, 4.0}); in TEST_F()
133 auto const2 = ConstantR1<float>(&builder, {2.0, 4.0, 6.0, 8.0}); in TEST_F()
162 ConstantR1<float>(&builder, {1.0, 2.0, 3.0, 4.0}); in TEST_F()
188 auto const1 = ConstantR1<float>(&builder, {1.0, 2.0, 3.0, 4.0}); in XLA_TEST_F()
[all …]
Daxpy_simple_test.cc33 auto x = ConstantR1<float>( in TEST_F()
46 auto x = ConstantR1<float>(&builder, {}); in XLA_TEST_F()
47 auto y = ConstantR1<float>(&builder, {}); in XLA_TEST_F()
58 auto x = ConstantR1<float>( in TEST_F()
60 auto y = ConstantR1<float>( in TEST_F()
Dselect_and_scatter_test.cc206 const auto operand = ConstantR1<float>(&builder_, {}); in XLA_TEST_F()
207 const auto source = ConstantR1<float>(&builder_, {}); in XLA_TEST_F()
217 ConstantR1<float>(&builder_, {1.f, 9.f, 3.f, 7.f, 5.f, 6.f}); in XLA_TEST_F()
218 const auto source = ConstantR1<float>(&builder_, {34.f, 42.f}); in XLA_TEST_F()
228 const auto operand = ConstantR1<int32>(&builder_, {-1, 0, 6, 4, -4, 10}); in XLA_TEST_F()
229 const auto source = ConstantR1<int32>(&builder_, {-10, 20}); in XLA_TEST_F()
239 const auto operand = ConstantR1<int32>(&builder_, {1, 9, 3, 7, 5, 6}); in XLA_TEST_F()
240 const auto source = ConstantR1<int32>(&builder_, {34, 42, 53, 19}); in XLA_TEST_F()
439 const auto operand = ConstantR1<float>(&builder_, {1, 2, 3, 100, 3, 2, 1}); in XLA_TEST_F()
440 const auto source = ConstantR1<float>(&builder_, {34, 42, 53, 19}); in XLA_TEST_F()
[all …]
Dbfloat16_test.cc92 auto scale = ConstantR1<bfloat16>( in XLA_TEST_F()
95 auto offset = ConstantR1<bfloat16>( in XLA_TEST_F()
124 auto scale = ConstantR1<bfloat16>( in XLA_TEST_F()
127 auto mean = ConstantR1<bfloat16>( in XLA_TEST_F()
130 auto var = ConstantR1<bfloat16>( in XLA_TEST_F()
Dconstants_test.cc46 ConstantR1<float>(&builder, {}); in TEST_F()
55 ConstantR1<float>(&builder, constant); in TEST_F()
64 ConstantR1<int32>(&builder, constant); in TEST_F()
73 ConstantR1<uint32>(&builder, constant); in TEST_F()
82 ConstantR1<float>(&builder, constant); in TEST_F()
92 ConstantR1<float>(&builder, constant); in TEST_F()
Ddeallocation_test.cc69 ConstantR1<float>(&builder, {1.0, 2.0, 3.0, 4.0}); in TEST_F()
82 ConstantR1<float>(&builder, {}); in TEST_F()
96 ConstantR1<float>(&builder, {1.0, 2.0, 3.0})}); in XLA_TEST_F()
127 ConstantR1<float>(&builder, {1.0, 2.0, 3.0})}); in XLA_TEST_F()
128 Tuple(&builder, {inner_tuple, ConstantR1<float>(&builder, {0.123, 0.456})}); in XLA_TEST_F()
Dwhile_test.cc143 Reduce(ConstantR1<int32>(&builder, 2, 1), ConstantR0<int32>(&builder, 0), in XLA_TEST_F()
218 auto input = ConstantR1<float>(&builder, {}); in XLA_TEST_F()
225 auto init = ConstantR1<float>(&builder, {}); in XLA_TEST_F()
272 auto input = ConstantR1<float>(&builder, 8, 0.125f); in XLA_TEST_F()
279 auto init = ConstantR1<float>(&builder, 8, 0.f); in XLA_TEST_F()
332 auto input = ConstantR1<float>(&builder, 8, 0.125f); in XLA_TEST_F()
339 auto init = ConstantR1<float>(&builder, 8, 0.f); in XLA_TEST_F()
392 ConstantR1<float>(&builder, 3, 1.f), in XLA_TEST_F()
393 ConstantR1<float>(&builder, 3, 2.f), in XLA_TEST_F()
394 ConstantR1<float>(&builder, 3, 3.f)}); in XLA_TEST_F()
[all …]
Dlocal_client_allocation_test.cc41 auto x = ConstantR1<float>(&builder, {0.0f, 1.0f, 2.0f}); in XLA_TEST_F()
42 auto y = ConstantR1<float>(&builder, {2.0f, 3.0f, 4.0f}); in XLA_TEST_F()
77 auto x = ConstantR1<float>(&builder, {0.0f, 1.0f, 2.0f}); in XLA_TEST_F()
78 auto y = ConstantR1<float>(&builder, {2.0f, 3.0f, 4.0f}); in XLA_TEST_F()
Dbroadcast_simple_test.cc150 Broadcast(ConstantR1<float>(&b, {1, 2, 3}), {2});
164 BroadcastInDim(ConstantR1<float>(&b, {1, 2}), {2, 2}, {1});
177 BroadcastInDim(ConstantR1<float>(&b, {1, 2}), {2, 2}, {0});
226 BroadcastInDim(ConstantR1<float>(&b, {1, 2}), {3, 2}, {1});
268 Broadcast(ConstantR1<float>(&b, {}), {2}); in XLA_TEST_F()
276 Broadcast(ConstantR1<float>(&b, {1, 2, 3}), {0});
651 auto r1 = ConstantR1<float>(&b, {10, 20}); in XLA_TEST_F()
664 auto r1 = ConstantR1<float>(&b, {10, 20}); in XLA_TEST_F()
677 auto r1 = ConstantR1<float>(&b, {10, 20}); in XLA_TEST_F()
690 auto r1_0 = ConstantR1<float>(&b, {1000, 2000}); in XLA_TEST_F()
[all …]
Dpred_test.cc82 ConstantR1<bool>(&builder, {true, false, false, true}); in TEST_F()
98 auto a = ConstantR1<bool>(&builder, {true, false}); in TEST_F()
105 auto a = ConstantR1<bool>(&builder, {false, false}); in TEST_F()
112 auto a = ConstantR1<bool>(&builder, {}); in TEST_F()
Dvector_ops_reduce_test.cc59 auto x = ConstantR1<float>( in TEST_F()
73 auto x = ConstantR1<float>(&builder_, input); in TEST_F()
84 auto x = ConstantR1<float>( in TEST_F()
95 auto x = ConstantR1<float>( in TEST_F()
Dconditional_test.cc399 auto operand1 = ConstantR1<float>(&builder, {24.0f, 56.0f}); in XLA_TEST_F()
400 auto operand2 = ConstantR1<float>(&builder, {10.0f, 11.0f}); in XLA_TEST_F()
418 auto operand1 = ConstantR1<float>(&builder, {24.0f, 56.0f}); in XLA_TEST_P()
419 auto operand2 = ConstantR1<float>(&builder, {10.0f, 11.0f}); in XLA_TEST_P()
455 auto operand1 = ConstantR1<float>(&builder, {24.0f, 56.0f}); in XLA_TEST_F()
456 auto operand2 = ConstantR1<float>(&builder, {10.0f, 11.0f}); in XLA_TEST_F()
488 Tuple(&builder, {ConstantR1<float>(&builder, {12.2f, 15.8f}), in XLA_TEST_F()
489 ConstantR1<float>(&builder, {25.6f, 29.2f})}); in XLA_TEST_F()
508 auto true_array = ConstantR1<float>(&true_builder, {12.8f, 14.6f}); in XLA_TEST_F()
519 auto false_array = ConstantR1<float>(&false_builder, {26.4f, 32.6f}); in XLA_TEST_F()
[all …]
Dbatch_normalization_test.cc106 auto y = ConstantR1<float>(&builder, {3.14, 4.25}); in XLA_TEST_P()
150 auto set_means = ConstantR1<float>(&builder, {2.f, 4.2f}); in XLA_TEST_P()
163 auto variance = ConstantR1<float>(&builder, {6.f, .02f}); in XLA_TEST_P()
177 auto gamma = ConstantR1<float>(&builder, {1.0, 1.0}); in XLA_TEST_P()
178 auto beta = ConstantR1<float>(&builder, {0.0, 0.0}); in XLA_TEST_P()
196 auto epsilon2 = ConstantR1<float>(&builder, {kEpsilon, kEpsilon}); in XLA_TEST_P()
238 auto scale = ConstantR1<float>(&builder, {2.0f, 3.0f}); in XLA_TEST_P()
240 auto offset = ConstantR1<float>(&builder, {1.0f, 2.0f}); in XLA_TEST_P()
262 auto scale = ConstantR1<float>(&builder, {2.0f, 3.0f}); in XLA_TEST_P()
264 auto offset = ConstantR1<float>(&builder, {1.0f, 2.0f}); in XLA_TEST_P()
[all …]
Dcompute_constant_test.cc159 Add(ConstantR1<float>(&b, {1.0f}), ConstantR1<float>(&b, {1.0f})); in TEST_F()
220 Add(ConstantR1<int32>(&b, {1, 2}), ConstantR1<int32>(&b, {3, 4})); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dmath_test.cc66 ConstantR1<T>(&b, { in TestIsInfOrNan()
229 auto x = ConstantR1<float>( in XLA_TEST_F()
240 auto x = ConstantR1<float>( in XLA_TEST_F()
252 auto x = ConstantR1<float>(&builder, {0.0, -0.0}); in XLA_TEST_F()
260 auto x = ConstantR1<float>(&builder, {16.0, 1.0, 1024.0, 0.16, 0.2, 12345}); in XLA_TEST_F()
269 auto x = ConstantR1<float>(&builder, {1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 0.5, 1.5, in XLA_TEST_F()
298 auto x = ConstantR1<half>(&b, { in XLA_TEST_F()
315 auto x = ConstantR1<float>(&builder, {1.0, 0.5, 1 / 3.0, 0.25, 1 / 6.0, 0.125, in XLA_TEST_F()
345 auto x = ConstantR1<float>( in XLA_TEST_F()
357 auto x = ConstantR1<std::complex<float>>(&b, {{0, 0}}); in XLA_TEST_F()
[all …]
Dsorting_test.cc34 ConstantR1<float>(&builder, {0.0, 1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0}); in XLA_TEST_F()
42 ConstantR1<float>(&builder, {7.0, 6.0, 5.0, 4.0, 3.0, 2.0, 1.0, 0.0}); in XLA_TEST_F()
50 auto x_rev = ConstantR1<int>(&builder, {std::numeric_limits<int>::min(), in XLA_TEST_F()
59 auto x_rev = ConstantR1<int>(&builder, {std::numeric_limits<int>::min(), in XLA_TEST_F()
77 auto x = ConstantR1<float>(&builder, inputs); in XLA_TEST_F()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dresampler_ops.cc203 xla::ConstantR1<float>(ctx->builder(), {-1, -1}), warp_type), in BoundSamples()
208 xla::ConstantR1<float>( in BoundSamples()
312 auto shifting_value = xla::ConstantR1<int32>( in CalculateGradData()
385 auto shifting_value = xla::ConstantR1<int32>( in CalculateGradWarp()
413 xla::ConstantR1<float>(ctx->builder(), {0, 0, -1, 1}), data_type), in CalculateGradWarp()
421 xla::ConstantR1<float>(ctx->builder(), {-1, 1, 0, 0}), data_type), in CalculateGradWarp()
429 xla::ConstantR1<float>(ctx->builder(), {0, -1, 0, 1}), data_type), in CalculateGradWarp()
437 xla::ConstantR1<float>(ctx->builder(), {-1, 0, 1, 0}), data_type), in CalculateGradWarp()
560 xla::ConstantR1<float>( in Compile()

12