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

1 2 3 4 5

  /external/tensorflow/tensorflow/compiler/xla/
literal_util.cc 108 return LiteralUtil::CreateR0<uint8>(0);
110 return LiteralUtil::CreateR0<uint16>(0);
112 return LiteralUtil::CreateR0<uint32>(0);
114 return LiteralUtil::CreateR0<uint64>(0);
116 return LiteralUtil::CreateR0<int8>(0);
118 return LiteralUtil::CreateR0<int16>(0);
120 return LiteralUtil::CreateR0<int32>(0);
122 return LiteralUtil::CreateR0<int64>(0);
124 return LiteralUtil::CreateR0<half>(static_cast<half>(0.0f));
126 return LiteralUtil::CreateR0<bfloat16>(static_cast<bfloat16>(0.0f))
    [all...]
literal_test.cc 100 auto true_lit = LiteralUtil::CreateR0<bool>(true);
103 auto false_lit = LiteralUtil::CreateR0<bool>(false);
106 auto u32_lit = LiteralUtil::CreateR0<uint32>(42);
109 auto s32_lit = LiteralUtil::CreateR0<int32>(-999);
112 auto f32_lit = LiteralUtil::CreateR0<float>(3.14f);
115 auto f16_lit = LiteralUtil::CreateR0<half>(static_cast<half>(0.5f));
118 auto c64_lit = LiteralUtil::CreateR0<complex64>({3.14f, 2.78f});
121 auto c128_lit = LiteralUtil::CreateR0<complex128>({3.14f, 2.78f});
124 auto bf16_lit = LiteralUtil::CreateR0<bfloat16>(static_cast<bfloat16>(0.5f));
129 LiteralUtil::CreateR0<bfloat16>(static_cast<bfloat16>(3.14f))
    [all...]
  /external/tensorflow/tensorflow/compiler/tf2xla/lib/
util.cc 70 literal = xla::LiteralUtil::CreateR0<uint8>(value);
73 literal = xla::LiteralUtil::CreateR0<uint32>(value);
76 literal = xla::LiteralUtil::CreateR0<uint64>(value);
79 literal = xla::LiteralUtil::CreateR0<int8>(value);
82 literal = xla::LiteralUtil::CreateR0<int32>(value);
85 literal = xla::LiteralUtil::CreateR0<int64>(value);
88 literal = xla::LiteralUtil::CreateR0<float>(value);
91 literal = xla::LiteralUtil::CreateR0<double>(value);
94 literal = xla::LiteralUtil::CreateR0<complex64>(value);
97 literal = xla::LiteralUtil::CreateR0<complex128>(value)
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/tests/
literal_test_util_test.cc 35 LiteralUtil::CreateR0<int32>(42),
36 LiteralUtil::CreateR0<int32>(64),
47 LiteralUtil::CreateR0<int32>(42),
48 LiteralUtil::CreateR0<int32>(64),
51 LiteralUtil::CreateR0<int32>(64),
52 LiteralUtil::CreateR0<int32>(42),
61 auto two = LiteralUtil::CreateR0<float>(2);
62 auto four = LiteralUtil::CreateR0<float>(4);
token_hlo_test.cc 79 HloInstruction::CreateConstant(LiteralUtil::CreateR0<int32>(42)));
188 auto arg = LiteralUtil::CreateR0<bool>(true);
198 auto arg = LiteralUtil::CreateR0<bool>(false);
226 auto p0 = LiteralUtil::CreateR0<float>(10.0);
227 auto p1 = LiteralUtil::CreateR0<float>(3.0);
228 auto expected = LiteralUtil::CreateR0<float>(-156.0);
247 auto p0 = LiteralUtil::CreateR0<float>(10.0);
248 auto expected = LiteralUtil::CreateR0<float>(420.0);
round_trip_transfer_test.cc 48 RoundTripTest(LiteralUtil::CreateR0<int32>(42));
52 RoundTripTest(LiteralUtil::CreateR0<float>(42.0));
127 {LiteralUtil::CreateR0<float>(1.0), LiteralUtil::CreateR1<int>({2, 3})}));
compilation_cache_test.cc 52 LiteralUtil::CreateR0<float>(expected_result), result, error_spec_));
90 client_->TransferToServer(LiteralUtil::CreateR0<float>(42.0f))
93 client_->TransferToServer(LiteralUtil::CreateR0<float>(123.0f))
96 client_->TransferToServer(LiteralUtil::CreateR0<float>(456.0f))
pad_test.cc 97 AddParam(LiteralUtil::CreateR0<float>(0.1), &b), padding_config);
112 AddParam(LiteralUtil::CreateR0<float>(0.1), &b), padding_config);
127 AddParam(LiteralUtil::CreateR0<float>(0.1), &b), padding_config);
135 AddParam(LiteralUtil::CreateR0<float>(1.5), &b),
151 Pad(AddParam(*input, &b), AddParam(LiteralUtil::CreateR0<float>(1.5), &b),
171 AddParam(LiteralUtil::CreateR0<float>(pad_value), &b),
214 AddParam(LiteralUtil::CreateR0<float>(pad_value), &b), padding_config);
260 AddParam(LiteralUtil::CreateR0<float>(pad_value), &b), padding_config);
334 Pad(input, AddParam(LiteralUtil::CreateR0<float>(0.0f), &b), padding_config);
356 Pad(input, AddParam(LiteralUtil::CreateR0<float>(3.14f), &b), padding_config)
    [all...]
multioutput_fusion_test.cc 72 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(8.0f)));
117 Literal literal_r0 = LiteralUtil::CreateR0<float>(-9.0f);
214 LiteralUtil::MakeTupleOwned(LiteralUtil::CreateR0<int32>(42)),
215 LiteralUtil::CreateR0<float>(1.0)),
216 LiteralUtil::MakeTupleOwned(LiteralUtil::CreateR0<float>(3.0),
217 LiteralUtil::CreateR0<int32>(4)));
220 LiteralUtil::MakeTupleOwned(LiteralUtil::CreateR0<int32>(42)), result));
501 auto init1 = LiteralUtil::CreateR0<float>(5);
502 auto init2 = LiteralUtil::CreateR0<float>(6);
call_test.cc 80 auto constant = ConstantLiteral(&builder, LiteralUtil::CreateR0<float>(42.0));
135 client_->TransferToServer(LiteralUtil::CreateR0<float>(1.0f)));
142 auto elem = LiteralUtil::CreateR0<float>(42.0);
multiple_devices_on_host_test.cc 32 ConstantLiteral(&b, LiteralUtil::CreateR0<int32>(1)),
88 LiteralUtil::CreateR0<int32>(device_ordinal * 100), device_ordinal));
96 EXPECT_EQ(outfeed, LiteralUtil::CreateR0<int32>(device_ordinal * 100 + 1));
  /external/tensorflow/tensorflow/compiler/xla/service/
conditional_simplifier_test.cc 58 HloInstruction::CreateConstant(LiteralUtil::CreateR0<int32>(1)));
76 HloInstruction::CreateConstant(LiteralUtil::CreateR0<int32>(42)));
85 HloInstruction::CreateConstant(LiteralUtil::CreateR0<bool>(false)));
89 HloInstruction::CreateConstant(LiteralUtil::CreateR0<int32>(1)));
111 HloInstruction::CreateConstant(LiteralUtil::CreateR0<bool>(true)));
128 HloInstruction::CreateConstant(LiteralUtil::CreateR0<bool>(true))),
hlo_alias_analysis_test.cc 121 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(1.0)));
123 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(2.0)));
357 HloInstruction::CreateConstant(LiteralUtil::CreateR0<bool>(false)));
425 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(1.0)));
427 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(2.0)));
465 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(1.0)));
467 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(2.0)));
545 HloInstruction::CreateConstant(LiteralUtil::CreateR0<bool>(false)));
551 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(1.0)));
553 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(2.0)))
    [all...]
call_inliner_test.cc 50 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(24.0f)));
52 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(42.0f)));
89 HloInstruction::CreateConstant(LiteralUtil::CreateR0<bool>(false)));
103 HloInstruction::CreateConstant(LiteralUtil::CreateR0<bool>(false)));
129 HloInstruction::CreateConstant(LiteralUtil::CreateR0<bool>(false)));
151 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(42.0)));
hlo_dce_test.cc 54 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(42.0f)));
56 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(123.0f)));
75 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(42.0f)));
128 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(42.0f)));
130 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(123.0f)));
225 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(42.0f)));
347 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(0))),
352 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(0)));
388 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(0))),
398 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(0)))
    [all...]
hlo_reachability_test.cc 42 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(0.0f)));
44 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(0.0f)));
46 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(0.0f)));
48 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(0.0f)));
50 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(0.0f)));
104 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(1.0f)));
106 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(2.0f)));
hlo_computation_test.cc 128 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(42.0f)));
139 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(42.0f)));
154 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(42.0f)));
173 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(42.0f)));
175 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(42.0f)));
177 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(42.0f)));
179 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(42.0f)));
191 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(42.0f)));
193 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(42.0f)));
195 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(42.0f)))
    [all...]
flatten_call_graph_test.cc 83 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(0.0f)));
161 HloInstruction::CreateConstant(LiteralUtil::CreateR0<bool>(false)));
172 HloInstruction::CreateConstant(LiteralUtil::CreateR0<bool>(false)));
236 HloInstruction::CreateConstant(LiteralUtil::CreateR0<bool>(true)));
238 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(56.0f)));
240 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(12.0f)));
hlo_cse_test.cc 56 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(42.0f)));
58 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(42.0f)));
75 auto expected = LiteralUtil::CreateR0<float>(84.0);
147 HloInstruction::CreateConstant(LiteralUtil::CreateR0<uint32>(42))));
149 HloInstruction::CreateConstant(LiteralUtil::CreateR0<int32>(42))));
151 HloInstruction::CreateConstant(LiteralUtil::CreateR0<uint64>(42.0))));
153 HloInstruction::CreateConstant(LiteralUtil::CreateR0<int64>(42.0))));
155 HloInstruction::CreateConstant(LiteralUtil::CreateR0<double>(42.0))));
157 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(42.0f))));
160 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(42.0f))))
    [all...]
tuple_points_to_analysis_test.cc 126 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(1.0)));
128 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(2.0)));
179 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(1.0)));
181 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(2.0)));
186 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(3.0)));
240 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(1.0)));
242 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(2.0)));
247 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(3.0)));
271 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(1.0)));
288 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(1.0)))
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
infeed_test.cc 55 TestInfeedRoundTrip(LiteralUtil::CreateR0<bool>(true));
103 LiteralUtil::CreateR0<bool>(false)}));
116 LiteralUtil::CreateR0<int32>(5)}));
  /external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/
cpu_infeed_test.cc 61 TestInfeedRoundTrip(LiteralUtil::CreateR0<bool>(true));
102 LiteralUtil::CreateR0<bool>(false)}));
254 LiteralUtil::CreateR0<bool>(true)})));
257 LiteralUtil::CreateR0<bool>(true)})));
260 LiteralUtil::CreateR0<bool>(true)})));
263 LiteralUtil::CreateR0<bool>(false)})));
279 LiteralUtil::CreateR0<bool>(true)})));
282 LiteralUtil::CreateR0<bool>(false)})));
285 LiteralUtil::CreateR0<bool>(true)})));
  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
cudnn_batchnorm_rewriter.cc 72 LiteralUtil::CreateR0(batch_norm->epsilon())));
75 LiteralUtil::CreateR0(batch_norm->feature_index())));
108 LiteralUtil::CreateR0(batch_norm->epsilon())));
111 LiteralUtil::CreateR0(batch_norm->feature_index())));
135 LiteralUtil::CreateR0<float>(-2))),
177 LiteralUtil::CreateR0(batch_norm->epsilon())));
180 LiteralUtil::CreateR0(batch_norm->feature_index())));
  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
cpu_copy_insertion_test.cc 77 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(123.0)));
84 HloInstruction::CreateConstant(LiteralUtil::CreateR0<bool>(false)));
117 HloInstruction::CreateConstant(LiteralUtil::CreateR0<float>(123.0)));
  /external/tensorflow/tensorflow/compiler/tf2xla/
tf2xla_test.cc 78 auto x_literal = xla::LiteralUtil::CreateR0<int32>(10);
79 auto y_literal = xla::LiteralUtil::CreateR0<int32>(32);

Completed in 1452 milliseconds

1 2 3 4 5