/external/tensorflow/tensorflow/compiler/xla/service/ |
while_util.cc | 45 HloInstruction::CreateConstant(LiteralUtil::CreateR0<bool>(false))); 161 HloInstruction::CreateConstant(LiteralUtil::CreateR0<int32>(trip_count))); 182 HloInstruction::CreateConstant(LiteralUtil::CreateR0<int32>(1))); 210 HloInstruction::CreateConstant(LiteralUtil::CreateR0<int32>(0)));
|
dynamic_padder_test.cc | 85 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(0.0))); 198 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(0.0)));
|
algebraic_simplifier_test.cc | 63 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(0.0f))); 286 HloInstruction::CreateConstant(LiteralUtil::CreateR0<int32>(0))); 307 HloInstruction::CreateConstant(LiteralUtil::CreateR0<bool>(true))); 329 HloInstruction::CreateConstant(LiteralUtil::CreateR0<bool>(false))); 368 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(0.0f))); 408 HloInstruction::CreateConstant(LiteralUtil::CreateR0(42.0f))); 429 HloInstruction::CreateConstant(LiteralUtil::CreateR0(42.0f))); 431 HloInstruction::CreateConstant(LiteralUtil::CreateR0(3.14159f))); 456 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(0.0f))); 491 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(0.0f))) [all...] |
hlo_instruction_test.cc | 253 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(1.1f))); 298 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(1.1f))); 338 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(1.0))); 387 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(0.0f))); 389 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(1.1f))); 630 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(1.1f))); 646 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(1.1f))); 648 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(42.1f))); 665 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(1.1f))); 686 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(1.1f))) [all...] |
buffer_assignment_test.cc | 160 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(1.0))); 188 HloInstruction::CreateConstant(LiteralUtil::CreateR0<int>(4))); 214 HloInstruction::CreateConstant(LiteralUtil::CreateR0<int>(1))); 337 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(1.0))); 386 HloInstruction::CreateConstant(LiteralUtil::CreateR0<int>(1))); [all...] |
hlo_graph_dumper_test.cc | 116 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(-42)));
|
hlo_rematerialization_test.cc | 135 HloInstruction::CreateConstant(LiteralUtil::CreateR0<bool>(true))); 234 HloInstruction::CreateConstant(LiteralUtil::CreateR0<bool>(true))); 270 HloInstruction::CreateConstant(LiteralUtil::CreateR0<bool>(true))); 302 HloInstruction::CreateConstant(LiteralUtil::CreateR0<bool>(true)));
|
hlo_constant_folding_test.cc | 46 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(42.0f))); 68 HloInstruction::CreateConstant(LiteralUtil::CreateR0<int64>(42)));
|
hlo_module_test.cc | 46 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(42.0f))); 130 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(42.0f)));
|
pattern_matcher_test.cc | 688 SetName("c", HloInstruction::CreateConstant(LiteralUtil::CreateR0(0))); 757 SetName("c", HloInstruction::CreateConstant(LiteralUtil::CreateR0(10))), 763 SetName("c", HloInstruction::CreateConstant(LiteralUtil::CreateR0(2.25))), [all...] |
hlo_evaluator_test.cc | 201 auto low = LiteralUtil::CreateR0<float>(0.f); 203 auto high = LiteralUtil::CreateR0<float>(1.f); 333 auto operand = LiteralUtil::CreateR0<float>(-1.0f); 334 auto expected = LiteralUtil::CreateR0<float>(1.0f); 459 auto input_literal = LiteralUtil::CreateR0<int32>(111); 578 auto pad_value = LiteralUtil::CreateR0<int32>(kPadValue); 604 auto pad_value = LiteralUtil::CreateR0<float>(kPadValue); 648 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(2.718f))); 689 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(2.718f))); [all...] |
buffer_liveness_test.cc | 443 Literal elements0[] = {LiteralUtil::CreateR0<int64>(0), 444 LiteralUtil::CreateR0<int64>(1)}; 446 Literal element1 = LiteralUtil::CreateR0<int64>(3); 641 HloInstruction::CreateConstant(LiteralUtil::CreateR0<int32>(2))); [all...] |
dynamic_dimension_inference_test.cc | 192 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(0.0))); 224 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(0.0))); 491 HloInstruction::CreateConstant(LiteralUtil::CreateR0<bool>(false))); 583 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(0.0))); 635 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(0.0)));
|
/external/tensorflow/tensorflow/compiler/xla/tests/ |
reduce_window_test.cc | 76 auto init = CreateConstantFromLiteral(LiteralUtil::CreateR0<float>(0.0f), 112 CreateConstantFromLiteral(LiteralUtil::CreateR0<float>(0), &builder_); 127 CreateConstantFromLiteral(LiteralUtil::CreateR0<float>(42.0), &builder_); 129 CreateConstantFromLiteral(LiteralUtil::CreateR0<float>(1.0), &builder_); 133 ComputeAndCompareLiteral(&builder_, LiteralUtil::CreateR0<float>(43.0), {}, 316 CreateConstantFromLiteral(LiteralUtil::CreateR0<float>(8.0f), b.get())); 321 CreateConstantFromLiteral(LiteralUtil::CreateR0<float>(0.0f), &builder_), 632 CreateConstantFromLiteral(LiteralUtil::CreateR0(kInitValue), &b); [all...] |
literal_test_util.h | 158 EXPECT_TRUE(Equal(LiteralUtil::CreateR0<NativeT>(expected), actual)); 204 EXPECT_TRUE(Near(LiteralUtil::CreateR0<NativeT>(expected), actual, error));
|
while_test.cc | 400 auto expected_counter = LiteralUtil::CreateR0<int32>(N); 509 auto expected_counter = LiteralUtil::CreateR0<int32>(5); 556 auto expected_counter = LiteralUtil::CreateR0<int32>(5); 557 auto expected_predicate = LiteralUtil::CreateR0<bool>(true); 601 auto expected_counter = LiteralUtil::CreateR0<int32>(5); 602 auto expected_data = LiteralUtil::CreateR0<int32>(7); [all...] |
local_client_execute_test.cc | 71 auto x_value = LiteralToShapedBuffer(LiteralUtil::CreateR0<float>(42.0f)); 481 inner_tuple_elements.push_back(LiteralUtil::CreateR0<float>(i + j)); 526 Literal arg_literal = LiteralUtil::CreateR0<float>(123.0); [all...] |
check_execution_arity_test.cc | 88 auto f32_literal = LiteralUtil::CreateR0<float>(1.1f);
|
params_test.cc | 45 Literal param0_literal = LiteralUtil::CreateR0<float>(3.14159f); 155 Literal literal = LiteralUtil::CreateR0<float>(3.14159f); 268 Literal literal = LiteralUtil::CreateR0<float>(i); 367 Literal bool_literal = LiteralUtil::CreateR0<bool>(false);
|
copy_test.cc | 61 TestCopyOp(LiteralUtil::CreateR0<bool>(true)); 91 auto literal = LiteralUtil::CreateR0<float>(42.0);
|
concat_test.cc | 563 auto x_literal = LiteralUtil::CreateR0<float>(2.f); 564 auto y_literal = LiteralUtil::CreateR0<float>(3.f); 586 auto y_literal = LiteralUtil::CreateR0<float>(1.5f); 587 auto z_literal = LiteralUtil::CreateR0<float>(5.5f); 614 auto y_literal = LiteralUtil::CreateR0<float>(1.5f); 615 auto z_literal = LiteralUtil::CreateR0<float>(5.5f);
|
map_test.cc | 172 Literal param0_literal = LiteralUtil::CreateR0<float>(42.0); 516 Literal param0_literal = LiteralUtil::CreateR0<float>(2.0f); 517 Literal param1_literal = LiteralUtil::CreateR0<float>(5.0f); 543 Literal param0_literal = LiteralUtil::CreateR0<float>(2.0f); 544 Literal param1_literal = LiteralUtil::CreateR0<float>(5.0f); 568 Literal param0_literal = LiteralUtil::CreateR0<float>(10.0f);
|
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
index_ops_cpu.cc | 102 xla::ConstantLiteral(&b, xla::LiteralUtil::CreateR0<int32>(axis)));
|
/external/tensorflow/tensorflow/compiler/tf2xla/ |
xla_expression_test.cc | 89 EXPECT_TRUE(xla::LiteralTestUtil::Equal(xla::LiteralUtil::CreateR0<int32>(42),
|
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
instruction_fusion_test.cc | 37 HloInstruction::CreateConstant(LiteralUtil::CreateR0(5))); 57 HloInstruction::CreateConstant(LiteralUtil::CreateR0(5))); 77 HloInstruction::CreateConstant(LiteralUtil::CreateR0(5))); 96 HloInstruction::CreateConstant(LiteralUtil::CreateR0(5)));
|