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

1 23 4 5

  /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)));

Completed in 596 milliseconds

1 23 4 5