HomeSort by relevance Sort by last modified time
    Searched refs:CreateR0 (Results 26 - 50 of 101) sorted by null

12 3 4 5

  /external/tensorflow/tensorflow/compiler/xla/tests/
tuple_test.cc 55 {LiteralUtil::CreateR0<float>(constant_scalar),
70 {LiteralUtil::CreateR0<float>(constant_scalar1),
71 LiteralUtil::CreateR0<float>(constant_scalar2)});
92 {LiteralUtil::CreateR0<float>(constant_scalar),
106 {LiteralUtil::CreateR0<float>(7.0), LiteralUtil::CreateR1<float>({})});
222 {LiteralUtil::CreateR0<bool>(direction),
223 LiteralUtil::CreateR0<bool>(!direction)});
421 auto expected_s = LiteralUtil::CreateR0<float>(42.0);
482 {LiteralUtil::CreateR0<complex64>({1, 2}),
509 LiteralUtil::CreateR0<complex64>({123, 456})})
    [all...]
conditional_test.cc 477 LiteralUtil::MakeTupleFromSlices({LiteralUtil::CreateR0<float>(12.0f),
478 LiteralUtil::CreateR0<float>(25.0f)}),
534 {LiteralUtil::CreateR0<bool>(true),
535 LiteralUtil::CreateR0<float>(12.2f),
581 {LiteralUtil::CreateR0<float>(46.6f),
585 LiteralUtil::CreateR0<float>(9.3f)})}),
748 {LiteralUtil::CreateR0<float>(a), LiteralUtil::CreateR0<float>(b)}),
784 LiteralUtil::MakeTupleFromSlices({LiteralUtil::CreateR0<int32>(123),
785 LiteralUtil::CreateR0<int32>(-42)}))
    [all...]
transfer_manager_test.cc 72 Literal literal = LiteralUtil::CreateR0<uint32>(42);
177 {LiteralUtil::CreateR0<float>(123.0f),
208 {LiteralUtil::CreateR0<float>(123.0f),
245 LiteralUtil::CreateR0<complex64>(complex64(0.3f, -0.4f))});
273 {LiteralUtil::CreateR0<float>(123.0f),
279 {LiteralUtil::CreateR0<float>(456.0f),
llvm_compiler_test.cc 69 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(42.0)));
91 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(42.0)));
outfeed_in_nested_computation_test.cc 84 local_client_->TransferToInfeed(LiteralUtil::CreateR0<int32_t>(1)));
157 local_client_->TransferToInfeed(LiteralUtil::CreateR0<bool>(true)));
replay_test.cc 96 client_->TransferToServer(LiteralUtil::CreateR0<int32>(2))
99 client_->TransferToServer(LiteralUtil::CreateR0<int32>(3))
broadcast_test.cc 40 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(42.0)));
49 EXPECT_TRUE(LiteralTestUtil::Near(LiteralUtil::CreateR0<float>(42.0), result,
56 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(42.0)));
228 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(1.0f)));
custom_call_test.cc 86 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(42.0f)));
202 Literal arg0 = LiteralUtil::CreateR0<float>(7.f);
203 Literal arg1 = LiteralUtil::CreateR0<float>(42.f);
scatter_test.cc 661 Literal scatter_indices = LiteralUtil::CreateR0<int32>(1);
689 Literal scatter_indices = LiteralUtil::CreateR0<int32>(1);
690 Literal updates = LiteralUtil::CreateR0<int32>(25);
742 Literal operand = LiteralUtil::CreateR0<int32>(1);
744 Literal updates = LiteralUtil::CreateR0<int32>(2);
scalar_computations_test.cc 164 Literal a_literal = LiteralUtil::CreateR0<int64>(value);
228 Literal a_literal = LiteralUtil::CreateR0<float>(2.1f);
229 Literal b_literal = LiteralUtil::CreateR0<float>(5.5f);
230 Literal c_literal = LiteralUtil::CreateR0<float>(0.5f);
377 auto dividend_literal = LiteralUtil::CreateR0<uint32>(dividend);
378 auto divisor_literal = LiteralUtil::CreateR0<uint32>(divisor);
390 LiteralUtil::CreateR0<uint32>(dividend / divisor);
419 auto dividend_literal = LiteralUtil::CreateR0<uint32>(dividend);
420 auto divisor_literal = LiteralUtil::CreateR0<uint32>(divisor);
432 LiteralUtil::CreateR0<uint32>(dividend % divisor)
    [all...]
fusion_test.cc 288 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(2.0)));
342 LiteralTestUtil::Equal(LiteralUtil::CreateR0<int32>(5),
387 LiteralTestUtil::Equal(LiteralUtil::CreateR0<int32>(7),
395 HloInstruction::CreateConstant(LiteralUtil::CreateR0<int32>(7)));
410 HloInstruction::CreateConstant(LiteralUtil::CreateR0<int32>(7)));
417 LiteralTestUtil::Equal(LiteralUtil::CreateR0<int32>(7),
504 HloInstruction::CreateConstant(LiteralUtil::CreateR0<int32>(1)));
542 HloInstruction::CreateConstant(LiteralUtil::CreateR0<int32>(1)));
612 HloInstruction::CreateConstant(LiteralUtil::CreateR0<int32>(0)));
621 LiteralTestUtil::Equal(LiteralUtil::CreateR0<int32>(15)
    [all...]
gather_operation_test.cc 382 Literal start_indices = LiteralUtil::CreateR0<int32>(1);
402 Literal start_indices = LiteralUtil::CreateR0<int32>(1);
616 Literal operand = LiteralUtil::CreateR0<float>(1);
  /external/tensorflow/tensorflow/compiler/xla/service/
hlo_dataflow_analysis_test.cc 105 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(1.0)));
107 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(2.0)));
204 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(1.0)));
206 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(2.0)));
266 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(1.0)));
268 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(2.0)));
316 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(1.0)));
318 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(2.0)));
371 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(1.0)));
373 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(2.0)))
    [all...]
hlo_get_dimension_size_rewriter.cc 51 HloInstruction::CreateConstant(LiteralUtil::CreateR0<uint32>(size)));
map_inliner_test.cc 86 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(2.0f)));
159 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(1)));
161 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(4)));
176 auto expected = LiteralUtil::CreateR0<float>(4);
copy_insertion_test.cc 111 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(1.0)));
170 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(1.0)));
172 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(2.0)));
200 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(1.0)));
202 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(2.0)));
204 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(3.0)));
212 HloInstruction::CreateConstant(LiteralUtil::CreateR0<bool>(false)));
374 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(1.0)));
376 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(2.0)));
384 HloInstruction::CreateConstant(LiteralUtil::CreateR0<bool>(false)))
    [all...]
hlo_creation_utils_test.cc 141 evaluator.Evaluate(*module, {LiteralUtil::CreateR0<int32>(9)}));
201 evaluator.Evaluate(*module, {LiteralUtil::CreateR0<int32>(0)}));
220 evaluator.Evaluate(*module, {LiteralUtil::CreateR0<float>(0.0f)}));
hlo_schedule_test.cc 100 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(42.0)));
150 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(42.0)));
193 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(1.0)));
hlo_subcomputation_unification_test.cc 77 HloInstruction::CreateConstant(LiteralUtil::CreateR0<int32>(5)));
104 HloInstruction::CreateConstant(LiteralUtil::CreateR0<int32>(5)));
106 HloInstruction::CreateConstant(LiteralUtil::CreateR0<int32>(3)));
hlo_ordering_test.cc 61 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(42.0f)));
149 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(1.0)));
183 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(1.0)));
232 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(1.0)));
419 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(123.0f)));
467 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(123.0f)));
473 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(42.0f)));
batchnorm_expander.cc 117 HloInstruction::CreateConstant(LiteralUtil::CreateR0<uint32>(1)));
214 auto zero_literal = LiteralUtil::CreateR0(0.0f);
218 auto epsilon_literal = LiteralUtil::CreateR0(batch_norm->epsilon());
343 auto epsilon_literal = LiteralUtil::CreateR0(batch_norm->epsilon());
475 auto zero_literal = LiteralUtil::CreateR0(0.0f);
479 auto epsilon_literal = LiteralUtil::CreateR0(batch_norm->epsilon());
  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
gpu_layout_assignment_test.cc 141 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(1)));
144 LiteralUtil::CreateR0<int64>(kFeatureIndex)));
211 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(1)));
214 LiteralUtil::CreateR0<int64>(kFeatureIndex)));
289 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(1)));
292 LiteralUtil::CreateR0<int64>(kFeatureIndex)));
while_transformer_test.cc 41 HloInstruction::CreateConstant(LiteralUtil::CreateR0<int32>(limit)));
65 LiteralUtil::CreateR0<int32>(increment)));
92 LiteralUtil::CreateR0<int32>(ind_var_init)));
  /external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/
cpu_fusion_test.cc 109 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(2.0))),
175 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(0))),
185 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(2.0))),
301 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(42.0)));
  /external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
gpu_ldg_test.cc 123 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(0))),

Completed in 1297 milliseconds

12 3 4 5