Home
last modified time | relevance | path

Searched refs:cond_builder (Results 1 – 17 of 17) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dloops.cc49 std::unique_ptr<XlaBuilder> cond_builder = in WhileLoopHelper() local
52 auto parameter = Parameter(cond_builder.get(), 0, tuple_shape, "parameter"); in WhileLoopHelper()
55 condition_function(unpack_tuple(parameter, arity, cond_builder.get()), in WhileLoopHelper()
56 cond_builder.get()) in WhileLoopHelper()
59 TF_ASSIGN_OR_RETURN(auto cond, cond_builder->Build()); in WhileLoopHelper()
88 XlaBuilder* cond_builder) -> StatusOr<XlaOp> { in ForEachIndex() argument
89 return Lt(values[0], ConstantR0WithType(cond_builder, num_iterations_type, in ForEachIndex()
Dself_adjoint_eig.cc226 XlaBuilder* cond_builder) -> StatusOr<XlaOp> { in WhileLoopFn() argument
234 xla::ConstantR0<bool>(cond_builder, false), in WhileLoopFn()
235 CreateScalarOrComputation(PRED, cond_builder)); in WhileLoopFn()
Dsvd.cc276 XlaBuilder* cond_builder) -> StatusOr<XlaOp> { in HouseHolderBidiagonalization() argument
605 XlaBuilder* cond_builder) -> StatusOr<XlaOp> { in WhileLoopFn() argument
613 xla::ConstantR0<bool>(cond_builder, false), in WhileLoopFn()
614 CreateScalarOrComputation(PRED, cond_builder)); in WhileLoopFn()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_alias_analysis_test.cc353 auto cond_builder = HloComputation::Builder("condition"); in TEST_F() local
354 auto cond_param = cond_builder.AddInstruction( in TEST_F()
356 cond_builder.AddInstruction( in TEST_F()
359 module_->AddEmbeddedComputation(cond_builder.Build()); in TEST_F()
541 auto cond_builder = HloComputation::Builder("condition"); in TEST_F() local
542 auto cond_param = cond_builder.AddInstruction( in TEST_F()
544 cond_builder.AddInstruction( in TEST_F()
547 module_->AddEmbeddedComputation(cond_builder.Build()); in TEST_F()
635 auto cond_builder = HloComputation::Builder("condition"); in TEST_F() local
636 cond_builder.AddInstruction( in TEST_F()
[all …]
Dhlo_rematerialization_test.cc230 auto cond_builder = HloComputation::Builder(TestName() + ".cond"); in TEST_F() local
231 cond_builder.AddInstruction( in TEST_F()
233 cond_builder.AddInstruction( in TEST_F()
236 module->AddEmbeddedComputation(cond_builder.Build()); in TEST_F()
266 auto cond_builder = HloComputation::Builder(TestName() + ".cond"); in TEST_F() local
267 cond_builder.AddInstruction( in TEST_F()
269 cond_builder.AddInstruction( in TEST_F()
272 module->AddEmbeddedComputation(cond_builder.Build()); in TEST_F()
298 auto cond_builder = HloComputation::Builder(TestName() + ".cond"); in TEST_F() local
299 cond_builder.AddInstruction( in TEST_F()
[all …]
Dhlo_ordering_test.cc139 auto cond_builder = HloComputation::Builder("condition"); in TEST_F() local
140 auto cond_param = cond_builder.AddInstruction( in TEST_F()
142 auto convert = cond_builder.AddInstruction(HloInstruction::CreateConvert( in TEST_F()
145 module->AddEmbeddedComputation(cond_builder.Build()); in TEST_F()
222 auto cond_builder = HloComputation::Builder("condition"); in TEST_F() local
223 auto cond_param = cond_builder.AddInstruction( in TEST_F()
225 auto convert = cond_builder.AddInstruction(HloInstruction::CreateConvert( in TEST_F()
228 module->AddEmbeddedComputation(cond_builder.Build()); in TEST_F()
Dheap_simulator_test.cc47 auto cond_builder = HloComputation::Builder("WhileCond"); in TEST_F() local
49 HloInstruction* cond_param = cond_builder.AddInstruction( in TEST_F()
51 HloInstruction* cond_iter = cond_builder.AddInstruction( in TEST_F()
53 HloInstruction* cond_data = cond_builder.AddInstruction( in TEST_F()
56 HloInstruction* cond_lt = cond_builder.AddInstruction( in TEST_F()
60 module->AddEmbeddedComputation(cond_builder.Build()); in TEST_F()
137 auto cond_builder = HloComputation::Builder("WhileCond"); in TEST_F() local
138 HloInstruction* cond_param = cond_builder.AddInstruction( in TEST_F()
141 cond_builder.AddInstruction(HloInstruction::CreateSlice( in TEST_F()
144 cond_builder.AddInstruction(HloInstruction::CreateReshape(r0f32, slice)); in TEST_F()
[all …]
Dcopy_insertion_test.cc1314 auto cond_builder = HloComputation::Builder("condition"); in TEST_F() local
1315 cond_builder.AddInstruction( in TEST_F()
1317 auto cond_constant = cond_builder.AddInstruction( in TEST_F()
1319 cond_builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
1322 module->AddEmbeddedComputation(cond_builder.Build()); in TEST_F()
1598 auto cond_builder = HloComputation::Builder("condition"); in TEST_F() local
1599 cond_builder.AddInstruction( in TEST_F()
1601 auto cond_constant = cond_builder.AddInstruction( in TEST_F()
1603 cond_builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
1606 module->AddEmbeddedComputation(cond_builder.Build()); in TEST_F()
[all …]
Dhlo_memory_scheduler_test.cc251 auto cond_builder = HloComputation::Builder("WhileCond"); in TEST_F() local
252 HloInstruction* cond_param = cond_builder.AddInstruction( in TEST_F()
255 cond_builder.AddInstruction(HloInstruction::CreateConstant( in TEST_F()
257 cond_builder.AddInstruction( in TEST_F()
260 auto cond_computation = module->AddEmbeddedComputation(cond_builder.Build()); in TEST_F()
Dhlo_dce_test.cc220 auto cond_builder = HloComputation::Builder(TestName() + "-cond"); in TEST_F() local
222 auto param = cond_builder.AddInstruction( in TEST_F()
224 auto constant = cond_builder.AddInstruction( in TEST_F()
226 cond_builder.AddInstruction( in TEST_F()
230 auto cond_computation = module->AddEmbeddedComputation(cond_builder.Build()); in TEST_F()
Dhlo_dataflow_analysis_test.cc500 auto cond_builder = HloComputation::Builder("condition"); in TEST_P() local
501 auto cond_param = cond_builder.AddInstruction( in TEST_P()
503 auto cond_constant = cond_builder.AddInstruction( in TEST_P()
506 module_->AddEmbeddedComputation(cond_builder.Build()); in TEST_P()
602 auto cond_builder = HloComputation::Builder("condition"); in TEST_P() local
603 cond_builder.AddInstruction( in TEST_P()
605 cond_builder.AddInstruction( in TEST_P()
608 module_->AddEmbeddedComputation(cond_builder.Build()); in TEST_P()
662 auto cond_builder = HloComputation::Builder("condition"); in TEST_P() local
663 cond_builder.AddInstruction( in TEST_P()
[all …]
Ddynamic_dimension_inference_test.cc487 auto cond_builder = HloComputation::Builder("condition"); in TEST_F() local
488 cond_builder.AddInstruction( in TEST_F()
490 cond_builder.AddInstruction( in TEST_F()
493 module_->AddEmbeddedComputation(cond_builder.Build()); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_copy_insertion_test.cc80 auto cond_builder = HloComputation::Builder("condition"); in TEST_F() local
81 cond_builder.AddInstruction( in TEST_F()
83 cond_builder.AddInstruction( in TEST_F()
86 module->AddEmbeddedComputation(cond_builder.Build()); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/tests/
Doutfeed_in_nested_computation_test.cc43 XlaBuilder cond_builder("loop_condition"); in XLA_TEST_F() local
44 XlaOp state_tuple = Parameter(&cond_builder, 0, state_tuple_shape, "state"); in XLA_TEST_F()
47 Gt(loop_counter, ConstantR0<int32_t>(&cond_builder, 0)); in XLA_TEST_F()
48 return cond_builder.Build(); in XLA_TEST_F()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dimage_ops.cc339 xla::XlaBuilder* cond_builder) const { in operator ()()
342 xla::Lt(row_idx, xla::ConstantR0<int32>(cond_builder, num_boxes)); in operator ()()
345 num_outputs_so_far, xla::ConstantR0<int32>(cond_builder, output_size)); in operator ()()
/external/deqp-deps/SPIRV-Tools/source/opt/
Dloop_peeling.cpp499 InstructionBuilder cond_builder( in PeelAfter() local
504 return cond_builder in PeelAfter()
505 .AddLessThan(cond_builder in PeelAfter()
/external/swiftshader/third_party/SPIRV-Tools/source/opt/
Dloop_peeling.cpp499 InstructionBuilder cond_builder( in PeelAfter() local
504 return cond_builder in PeelAfter()
505 .AddLessThan(cond_builder in PeelAfter()