Home
last modified time | relevance | path

Searched refs:CreateR0 (Results 1 – 25 of 109) sorted by relevance

12345

/external/tensorflow/tensorflow/compiler/xla/
Dliteral_util.cc128 return LiteralUtil::CreateR0<uint8>(0); in Zero()
130 return LiteralUtil::CreateR0<uint16>(0); in Zero()
132 return LiteralUtil::CreateR0<uint32>(0); in Zero()
134 return LiteralUtil::CreateR0<uint64>(0); in Zero()
136 return LiteralUtil::CreateR0<int8>(0); in Zero()
138 return LiteralUtil::CreateR0<int16>(0); in Zero()
140 return LiteralUtil::CreateR0<int32>(0); in Zero()
142 return LiteralUtil::CreateR0<int64>(0); in Zero()
144 return LiteralUtil::CreateR0<half>(static_cast<half>(0.0f)); in Zero()
146 return LiteralUtil::CreateR0<bfloat16>(static_cast<bfloat16>(0.0f)); in Zero()
[all …]
Dliteral_test.cc99 auto true_lit = LiteralUtil::CreateR0<bool>(true); in TEST_F()
102 auto false_lit = LiteralUtil::CreateR0<bool>(false); in TEST_F()
105 auto u32_lit = LiteralUtil::CreateR0<uint32>(42); in TEST_F()
108 auto s32_lit = LiteralUtil::CreateR0<int32>(-999); in TEST_F()
111 auto f32_lit = LiteralUtil::CreateR0<float>(3.14f); in TEST_F()
114 auto f16_lit = LiteralUtil::CreateR0<half>(static_cast<half>(0.5f)); in TEST_F()
117 auto c64_lit = LiteralUtil::CreateR0<complex64>({3.14f, 2.78f}); in TEST_F()
120 auto c128_lit = LiteralUtil::CreateR0<complex128>({3.14, 2.78}); in TEST_F()
123 auto bf16_lit = LiteralUtil::CreateR0<bfloat16>(static_cast<bfloat16>(0.5f)); in TEST_F()
128 LiteralUtil::CreateR0<bfloat16>(static_cast<bfloat16>(3.14f)); in TEST_F()
[all …]
/external/tensorflow/tensorflow/compiler/xla/tests/
Dliteral_test_util_test.cc35 LiteralUtil::CreateR0<int32>(42), in TEST()
36 LiteralUtil::CreateR0<int32>(64), in TEST()
43 LiteralUtil::CreateR0<complex64>({42.0, 64.0}), in TEST()
44 LiteralUtil::CreateR0<complex64>({64.0, 42.0}), in TEST()
51 LiteralUtil::CreateR0<complex128>({42.0, 64.0}), in TEST()
52 LiteralUtil::CreateR0<complex128>({64.0, 42.0}), in TEST()
59 LiteralUtil::CreateR0<complex64>({42.0, 64.0}), in TEST()
60 LiteralUtil::CreateR0<complex64>({64.0, 42.0}), in TEST()
63 LiteralUtil::CreateR0<complex64>({64.0, 42.0}), in TEST()
64 LiteralUtil::CreateR0<complex64>({42.0, 64.0}), in TEST()
[all …]
Dpad_test.cc97 AddParam(LiteralUtil::CreateR0<float>(0.1), &b), padding_config); in XLA_TEST_P()
112 AddParam(LiteralUtil::CreateR0<float>(0.1), &b), padding_config); in XLA_TEST_P()
127 AddParam(LiteralUtil::CreateR0<float>(0.1), &b), padding_config); in XLA_TEST_P()
135 AddParam(LiteralUtil::CreateR0<float>(1.5), &b), in XLA_TEST_P()
151 Pad(AddParam(*input, &b), AddParam(LiteralUtil::CreateR0<float>(1.5), &b), in TEST_P()
171 AddParam(LiteralUtil::CreateR0<float>(pad_value), &b), in TEST_P()
214 AddParam(LiteralUtil::CreateR0<float>(pad_value), &b), padding_config); in TEST_P()
260 AddParam(LiteralUtil::CreateR0<float>(pad_value), &b), padding_config); in XLA_TEST_P()
334 Pad(input, AddParam(LiteralUtil::CreateR0<float>(0.0f), &b), padding_config); in XLA_TEST_P()
356 Pad(input, AddParam(LiteralUtil::CreateR0<float>(3.14f), &b), padding_config); in XLA_TEST_P()
[all …]
Dtoken_hlo_test.cc78 HloInstruction::CreateConstant(LiteralUtil::CreateR0<int32>(42))); in XLA_TEST_F()
187 auto arg = LiteralUtil::CreateR0<bool>(true); in XLA_TEST_F()
197 auto arg = LiteralUtil::CreateR0<bool>(false); in XLA_TEST_F()
225 auto p0 = LiteralUtil::CreateR0<float>(10.0); in XLA_TEST_F()
226 auto p1 = LiteralUtil::CreateR0<float>(3.0); in XLA_TEST_F()
227 auto expected = LiteralUtil::CreateR0<float>(-156.0); in XLA_TEST_F()
246 auto p0 = LiteralUtil::CreateR0<float>(10.0); in XLA_TEST_F()
247 auto expected = LiteralUtil::CreateR0<float>(420.0); in XLA_TEST_F()
Dtuple_test.cc55 {LiteralUtil::CreateR0<float>(constant_scalar), in XLA_TEST_F()
70 {LiteralUtil::CreateR0<float>(constant_scalar1), in XLA_TEST_F()
71 LiteralUtil::CreateR0<float>(constant_scalar2)}); in XLA_TEST_F()
92 {LiteralUtil::CreateR0<float>(constant_scalar), in XLA_TEST_F()
106 {LiteralUtil::CreateR0<float>(7.0), LiteralUtil::CreateR1<float>({})}); in XLA_TEST_F()
222 {LiteralUtil::CreateR0<bool>(direction), in XLA_TEST_F()
223 LiteralUtil::CreateR0<bool>(!direction)}); in XLA_TEST_F()
421 auto expected_s = LiteralUtil::CreateR0<float>(42.0); in XLA_TEST_F()
482 {LiteralUtil::CreateR0<complex64>({1, 2}), in XLA_TEST_F()
509 LiteralUtil::CreateR0<complex64>({123, 456})}); in XLA_TEST_F()
[all …]
Dmultioutput_fusion_test.cc72 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(8.0f))); in RunTest2D()
117 Literal literal_r0 = LiteralUtil::CreateR0<float>(-9.0f); in RunTest2D()
216 LiteralUtil::MakeTupleOwned(LiteralUtil::CreateR0<int32>(42)), in XLA_TEST_F()
217 LiteralUtil::CreateR0<float>(1.0)), in XLA_TEST_F()
218 LiteralUtil::MakeTupleOwned(LiteralUtil::CreateR0<float>(3.0), in XLA_TEST_F()
219 LiteralUtil::CreateR0<int32>(4))); in XLA_TEST_F()
222 LiteralUtil::MakeTupleOwned(LiteralUtil::CreateR0<int32>(42)), result)); in XLA_TEST_F()
Dcompilation_cache_test.cc52 LiteralUtil::CreateR0<float>(expected_result), result, error_spec_)); in ExecuteComputationR0F32()
90 client_->TransferToServer(LiteralUtil::CreateR0<float>(42.0f)) in XLA_TEST_F()
93 client_->TransferToServer(LiteralUtil::CreateR0<float>(123.0f)) in XLA_TEST_F()
96 client_->TransferToServer(LiteralUtil::CreateR0<float>(456.0f)) in XLA_TEST_F()
Dround_trip_transfer_test.cc48 RoundTripTest(LiteralUtil::CreateR0<int32>(42)); in TEST_F()
52 RoundTripTest(LiteralUtil::CreateR0<float>(42.0)); in TEST_F()
127 {LiteralUtil::CreateR0<float>(1.0), LiteralUtil::CreateR1<int>({2, 3})})); in TEST_F()
/external/tensorflow/tensorflow/compiler/tf2xla/lib/
Dutil.cc70 literal = xla::LiteralUtil::CreateR0<uint8>(value); in IntegerLiteral()
73 literal = xla::LiteralUtil::CreateR0<uint16>(value); in IntegerLiteral()
76 literal = xla::LiteralUtil::CreateR0<uint32>(value); in IntegerLiteral()
79 literal = xla::LiteralUtil::CreateR0<uint64>(value); in IntegerLiteral()
82 literal = xla::LiteralUtil::CreateR0<int8>(value); in IntegerLiteral()
85 literal = xla::LiteralUtil::CreateR0<int16>(value); in IntegerLiteral()
88 literal = xla::LiteralUtil::CreateR0<int32>(value); in IntegerLiteral()
91 literal = xla::LiteralUtil::CreateR0<int64>(value); in IntegerLiteral()
94 literal = xla::LiteralUtil::CreateR0<float>(value); in IntegerLiteral()
97 literal = xla::LiteralUtil::CreateR0<double>(value); in IntegerLiteral()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/
Dcpu_infeed_test.cc61 TestInfeedRoundTrip(LiteralUtil::CreateR0<bool>(true)); in TEST_F()
102 LiteralUtil::CreateR0<bool>(false)})); in TEST_F()
254 LiteralUtil::CreateR0<bool>(true)}))); in TEST_F()
257 LiteralUtil::CreateR0<bool>(true)}))); in TEST_F()
260 LiteralUtil::CreateR0<bool>(true)}))); in TEST_F()
263 LiteralUtil::CreateR0<bool>(false)}))); in TEST_F()
279 LiteralUtil::CreateR0<bool>(true)}))); in TEST_F()
282 LiteralUtil::CreateR0<bool>(false)}))); in TEST_F()
285 LiteralUtil::CreateR0<bool>(true)}))); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_alias_analysis_test.cc121 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(1.0))); in TEST_F()
123 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(2.0))); in TEST_F()
357 HloInstruction::CreateConstant(LiteralUtil::CreateR0<bool>(false))); in TEST_F()
425 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(1.0))); in TEST_F()
427 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(2.0))); in TEST_F()
465 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(1.0))); in TEST_F()
467 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(2.0))); in TEST_F()
545 HloInstruction::CreateConstant(LiteralUtil::CreateR0<bool>(false))); in TEST_F()
551 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(1.0))); in TEST_F()
553 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(2.0))); in TEST_F()
[all …]
Dcall_inliner_test.cc50 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(24.0f))); in TEST_F()
52 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(42.0f))); in TEST_F()
89 HloInstruction::CreateConstant(LiteralUtil::CreateR0<bool>(false))); in TEST_F()
103 HloInstruction::CreateConstant(LiteralUtil::CreateR0<bool>(false))); in TEST_F()
129 HloInstruction::CreateConstant(LiteralUtil::CreateR0<bool>(false))); in TEST_F()
153 HloInstruction::CreateConstant(LiteralUtil::CreateR0<int32>(0))); in TEST_F()
160 HloInstruction::CreateConstant(LiteralUtil::CreateR0<int32>(0))); in TEST_F()
166 HloInstruction::CreateConstant(LiteralUtil::CreateR0<int32>(0))); in TEST_F()
191 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(42.0))); in TEST_F()
Dtuple_points_to_analysis_test.cc126 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(1.0))); in TEST_F()
128 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(2.0))); in TEST_F()
179 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(1.0))); in TEST_F()
181 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(2.0))); in TEST_F()
186 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(3.0))); in TEST_F()
240 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(1.0))); in TEST_F()
242 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(2.0))); in TEST_F()
247 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(3.0))); in TEST_F()
271 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(1.0))); in TEST_F()
288 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(1.0))); in TEST_F()
[all …]
Dhlo_computation_test.cc127 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(42.0f))); in TEST_F()
138 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(42.0f))); in TEST_F()
153 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(42.0f))); in TEST_F()
172 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(42.0f))); in TEST_F()
174 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(42.0f))); in TEST_F()
176 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(42.0f))); in TEST_F()
178 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(42.0f))); in TEST_F()
190 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(42.0f))); in TEST_F()
192 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(42.0f))); in TEST_F()
194 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(42.0f))); in TEST_F()
[all …]
Dhlo_cse_test.cc56 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(42.0f))); in TEST_F()
58 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(42.0f))); in TEST_F()
75 auto expected = LiteralUtil::CreateR0<float>(84.0); in TEST_F()
147 HloInstruction::CreateConstant(LiteralUtil::CreateR0<uint32>(42)))); in TEST_F()
149 HloInstruction::CreateConstant(LiteralUtil::CreateR0<int32>(42)))); in TEST_F()
151 HloInstruction::CreateConstant(LiteralUtil::CreateR0<uint64>(42.0)))); in TEST_F()
153 HloInstruction::CreateConstant(LiteralUtil::CreateR0<int64>(42.0)))); in TEST_F()
155 HloInstruction::CreateConstant(LiteralUtil::CreateR0<double>(42.0)))); in TEST_F()
157 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(42.0f)))); in TEST_F()
160 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(42.0f)))); in TEST_F()
[all …]
Dhlo_dataflow_analysis_test.cc105 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(1.0))); in TEST_P()
107 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(2.0))); in TEST_P()
204 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(1.0))); in TEST_P()
206 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(2.0))); in TEST_P()
266 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(1.0))); in TEST_P()
268 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(2.0))); in TEST_P()
316 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(1.0))); in TEST_P()
318 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(2.0))); in TEST_P()
371 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(1.0))); in TEST_P()
373 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(2.0))); in TEST_P()
[all …]
Ddynamic_padder_test.cc91 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(0.0))); in TEST_F()
204 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(0.0))); in TEST_F()
274 Literal dynamic_size = LiteralUtil::CreateR0<int32>(2); in XLA_TEST_F()
338 Literal dynamic_size = LiteralUtil::CreateR0<int32>(1); in XLA_TEST_F()
408 Literal expected = LiteralUtil::CreateR0<int32>(3); in XLA_TEST_F()
439 Literal dynamic_size = LiteralUtil::CreateR0<int32>(2); in XLA_TEST_F()
495 Literal expected = LiteralUtil::CreateR0<int32>(6); in XLA_TEST_F()
540 Literal expected = LiteralUtil::CreateR0<int32>(28); in XLA_TEST_F()
573 Literal expected = LiteralUtil::CreateR0<int32>(6); in XLA_TEST_F()
608 Literal expected = LiteralUtil::CreateR0<int32>(18); in XLA_TEST_F()
[all …]
Dhlo_dce_test.cc56 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(42.0f))); in TEST_F()
58 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(123.0f))); in TEST_F()
77 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(42.0f))); in TEST_F()
166 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(42.0f))); in TEST_F()
168 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(123.0f))); in TEST_F()
263 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(42.0f))); in TEST_F()
385 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(0))), in TEST_F()
390 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(0))); in TEST_F()
426 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(0))), in TEST_F()
436 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(0))), in TEST_F()
Dhlo_reachability_test.cc42 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(0.0f))); in TEST_F()
44 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(0.0f))); in TEST_F()
46 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(0.0f))); in TEST_F()
48 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(0.0f))); in TEST_F()
50 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(0.0f))); in TEST_F()
104 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(1.0f))); in TEST_F()
106 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(2.0f))); in TEST_F()
Dflatten_call_graph_test.cc83 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(0.0f))); in MakeConditionComputation()
161 HloInstruction::CreateConstant(LiteralUtil::CreateR0<bool>(false))); in TEST_F()
172 HloInstruction::CreateConstant(LiteralUtil::CreateR0<bool>(false))); in TEST_F()
236 HloInstruction::CreateConstant(LiteralUtil::CreateR0<bool>(true))); in TEST_F()
238 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(56.0f))); in TEST_F()
240 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(12.0f))); in TEST_F()
Dcopy_insertion_test.cc111 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(1.0))); in TEST_F()
170 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(1.0))); in TEST_F()
172 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(2.0))); in TEST_F()
200 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(1.0))); in TEST_F()
202 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(2.0))); in TEST_F()
204 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(3.0))); in TEST_F()
212 HloInstruction::CreateConstant(LiteralUtil::CreateR0<bool>(false))); in TEST_F()
374 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(1.0))); in TEST_F()
376 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(2.0))); in TEST_F()
384 HloInstruction::CreateConstant(LiteralUtil::CreateR0<bool>(false))); in TEST_F()
[all …]
Dhlo_creation_utils_test.cc156 evaluator.Evaluate(*module, {LiteralUtil::CreateR0<int32>(9)})); in TEST_F()
216 evaluator.Evaluate(*module, {LiteralUtil::CreateR0<int32>(0)})); in TEST_F()
235 evaluator.Evaluate(*module, {LiteralUtil::CreateR0<float>(0.0f)})); in TEST_F()
277 evaluator.Evaluate(*module, {LiteralUtil::CreateR0<int32>(0.0)})); in TEST_F()
296 evaluator.Evaluate(*module, {LiteralUtil::CreateR0<float>(0.0f)})); in TEST_F()
316 evaluator.Evaluate(*module, {LiteralUtil::CreateR0<int32>(0.0)})); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dinfeed_test.cc55 TestInfeedRoundTrip(LiteralUtil::CreateR0<bool>(true)); in TEST_F()
103 LiteralUtil::CreateR0<bool>(false)})); in TEST_F()
116 LiteralUtil::CreateR0<int32>(5)})); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_batchnorm_rewriter.cc87 LiteralUtil::CreateR0(batch_norm->epsilon()))); in HandleBatchNormInference()
90 LiteralUtil::CreateR0(batch_norm->feature_index()))); in HandleBatchNormInference()
139 LiteralUtil::CreateR0(batch_norm->epsilon()))); in HandleBatchNormTraining()
142 LiteralUtil::CreateR0(batch_norm->feature_index()))); in HandleBatchNormTraining()
180 LiteralUtil::CreateR0<float>(-2))), in HandleBatchNormTraining()
228 LiteralUtil::CreateR0(batch_norm->epsilon()))); in HandleBatchNormGrad()
231 LiteralUtil::CreateR0(batch_norm->feature_index()))); in HandleBatchNormGrad()

12345