OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
defs:cond_builder
(Results
1 - 10
of
10
) sorted by null
/external/tensorflow/tensorflow/compiler/tf2xla/lib/
while_loop.cc
48
std::unique_ptr<xla::ComputationBuilder>
cond_builder
=
local
51
auto parameter =
cond_builder
->Parameter(0, tuple_shape, "parameter");
55
condition_function(unpack_tuple(parameter, arity,
cond_builder
.get()),
56
cond_builder
.get()));
57
TF_RETURN_IF_ERROR(
cond_builder
->SetReturnValue(result));
59
TF_ASSIGN_OR_RETURN(auto cond,
cond_builder
->Build());
88
xla::ComputationBuilder*
cond_builder
)
90
return
cond_builder
->Lt(
92
IntegerLiteral(
cond_builder
, num_iterations_type, num_iterations));
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
cpu_copy_insertion_test.cc
80
auto
cond_builder
= HloComputation::Builder("condition");
local
81
cond_builder
.AddInstruction(
83
cond_builder
.AddInstruction(
86
module->AddEmbeddedComputation(
cond_builder
.Build());
/external/tensorflow/tensorflow/compiler/xla/service/
hlo_scheduling_test.cc
41
auto
cond_builder
= HloComputation::Builder("WhileCond");
local
43
HloInstruction* cond_param =
cond_builder
.AddInstruction(
45
HloInstruction* cond_iter =
cond_builder
.AddInstruction(
47
HloInstruction* cond_data =
cond_builder
.AddInstruction(
50
HloInstruction* cond_lt =
cond_builder
.AddInstruction(
54
module->AddEmbeddedComputation(
cond_builder
.Build());
hlo_dce_test.cc
221
auto
cond_builder
= HloComputation::Builder(TestName() + "-cond");
local
223
auto param =
cond_builder
.AddInstruction(
225
auto constant =
cond_builder
.AddInstruction(
227
cond_builder
.AddInstruction(HloInstruction::CreateBinary(
230
auto cond_computation = module->AddEmbeddedComputation(
cond_builder
.Build());
hlo_ordering_test.cc
185
auto
cond_builder
= HloComputation::Builder("condition");
local
186
auto cond_param =
cond_builder
.AddInstruction(
188
auto convert =
cond_builder
.AddInstruction(HloInstruction::CreateConvert(
191
module->AddEmbeddedComputation(
cond_builder
.Build());
248
auto
cond_builder
= HloComputation::Builder("condition");
local
249
auto cond_param =
cond_builder
.AddInstruction(
251
auto convert =
cond_builder
.AddInstruction(HloInstruction::CreateConvert(
254
module->AddEmbeddedComputation(
cond_builder
.Build());
hlo_rematerialization_test.cc
214
auto
cond_builder
= HloComputation::Builder(TestName() + ".cond");
local
215
cond_builder
.AddInstruction(
217
cond_builder
.AddInstruction(
220
module->AddEmbeddedComputation(
cond_builder
.Build());
253
auto
cond_builder
= HloComputation::Builder(TestName() + ".cond");
local
254
cond_builder
.AddInstruction(
256
cond_builder
.AddInstruction(
259
module->AddEmbeddedComputation(
cond_builder
.Build());
288
auto
cond_builder
= HloComputation::Builder(TestName() + ".cond");
local
289
cond_builder
.AddInstruction
[
all
...]
heap_simulator_test.cc
420
auto
cond_builder
= HloComputation::Builder("WhileCond");
local
421
HloInstruction* cond_param =
cond_builder
.AddInstruction(
423
HloInstruction* cond_iter =
cond_builder
.AddInstruction(
425
HloInstruction* cond_data =
cond_builder
.AddInstruction(
427
HloInstruction* cond_lt =
cond_builder
.AddInstruction(
431
tracker.module()->AddEmbeddedComputation(
cond_builder
.Build());
[
all
...]
hlo_alias_analysis_test.cc
345
auto
cond_builder
= HloComputation::Builder("condition");
local
346
auto cond_param =
cond_builder
.AddInstruction(
348
cond_builder
.AddInstruction(
351
module_->AddEmbeddedComputation(
cond_builder
.Build());
438
auto
cond_builder
= HloComputation::Builder("condition");
local
439
cond_builder
.AddInstruction(
441
cond_builder
.AddInstruction(
444
module_->AddEmbeddedComputation(
cond_builder
.Build());
497
auto
cond_builder
= HloComputation::Builder("condition");
498
cond_builder
.AddInstruction
607
auto
cond_builder
= HloComputation::Builder("condition");
local
753
auto
cond_builder
= HloComputation::Builder("condition");
local
849
auto
cond_builder
= HloComputation::Builder("condition");
local
[
all
...]
copy_insertion_test.cc
1304
auto
cond_builder
= HloComputation::Builder("condition");
local
1369
auto
cond_builder
= HloComputation::Builder("condition");
local
1429
auto
cond_builder
= HloComputation::Builder("condition");
local
1514
auto
cond_builder
= HloComputation::Builder("condition");
local
1576
auto
cond_builder
= HloComputation::Builder("condition");
local
[
all
...]
hlo_dataflow_analysis_test.cc
492
auto
cond_builder
= HloComputation::Builder("condition");
local
493
auto cond_param =
cond_builder
.AddInstruction(
495
auto cond_constant =
cond_builder
.AddInstruction(
498
module_->AddEmbeddedComputation(
cond_builder
.Build());
593
auto
cond_builder
= HloComputation::Builder("condition");
local
594
cond_builder
.AddInstruction(
596
cond_builder
.AddInstruction(
599
module_->AddEmbeddedComputation(
cond_builder
.Build());
652
auto
cond_builder
= HloComputation::Builder("condition");
local
653
cond_builder
.AddInstruction
779
auto
cond_builder
= HloComputation::Builder("condition");
local
1025
auto
cond_builder
= HloComputation::Builder("condition");
local
1319
auto
cond_builder
= HloComputation::Builder("condition");
local
[
all
...]
Completed in 216 milliseconds