| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | loops.cc | 49 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()
|
| D | self_adjoint_eig.cc | 226 XlaBuilder* cond_builder) -> StatusOr<XlaOp> { in WhileLoopFn() argument
234 xla::ConstantR0<bool>(cond_builder, false), in WhileLoopFn()
235 CreateScalarOrComputation(PRED, cond_builder)); in WhileLoopFn()
|
| D | svd.cc | 270 XlaBuilder* cond_builder) -> StatusOr<XlaOp> { in HouseHolderBidiagonalization() argument
595 XlaBuilder* cond_builder) -> StatusOr<XlaOp> { in WhileLoopFn() argument
603 tolerance_comparison, xla::ConstantR0<bool>(cond_builder, false), in WhileLoopFn()
604 CreateScalarOrComputation(PRED, cond_builder)); in WhileLoopFn()
|
| D | math.cc | 1655 XlaBuilder* cond_builder) -> StatusOr<XlaOp> { in LentzThompsonBarnettAlgorithm() argument
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_alias_analysis_test.cc | 347 auto cond_builder = HloComputation::Builder("condition"); in TEST_F() local
348 auto cond_param = cond_builder.AddInstruction( in TEST_F()
350 cond_builder.AddInstruction( in TEST_F()
353 module_->AddEmbeddedComputation(cond_builder.Build()); in TEST_F()
533 auto cond_builder = HloComputation::Builder("condition"); in TEST_F() local
534 auto cond_param = cond_builder.AddInstruction( in TEST_F()
536 cond_builder.AddInstruction( in TEST_F()
539 module_->AddEmbeddedComputation(cond_builder.Build()); in TEST_F()
627 auto cond_builder = HloComputation::Builder("condition"); in TEST_F() local
628 cond_builder.AddInstruction( in TEST_F()
[all …]
|
| D | hlo_rematerialization_test.cc | 150 auto cond_builder = HloComputation::Builder(TestName() + ".cond"); in TEST_F() local
151 cond_builder.AddInstruction( in TEST_F()
153 cond_builder.AddInstruction( in TEST_F()
156 module->AddEmbeddedComputation(cond_builder.Build()); in TEST_F()
186 auto cond_builder = HloComputation::Builder(TestName() + ".cond"); in TEST_F() local
187 cond_builder.AddInstruction( in TEST_F()
189 cond_builder.AddInstruction( in TEST_F()
192 module->AddEmbeddedComputation(cond_builder.Build()); in TEST_F()
218 auto cond_builder = HloComputation::Builder(TestName() + ".cond"); in TEST_F() local
219 cond_builder.AddInstruction( in TEST_F()
[all …]
|
| D | hlo_ordering_test.cc | 138 auto cond_builder = HloComputation::Builder("condition"); in TEST_F() local
139 auto cond_param = cond_builder.AddInstruction( in TEST_F()
141 auto convert = cond_builder.AddInstruction(HloInstruction::CreateConvert( in TEST_F()
144 module->AddEmbeddedComputation(cond_builder.Build()); in TEST_F()
221 auto cond_builder = HloComputation::Builder("condition"); in TEST_F() local
222 auto cond_param = cond_builder.AddInstruction( in TEST_F()
224 auto convert = cond_builder.AddInstruction(HloInstruction::CreateConvert( in TEST_F()
227 module->AddEmbeddedComputation(cond_builder.Build()); in TEST_F()
|
| D | hlo_live_range_test.cc | 269 auto cond_builder = HloComputation::Builder("WhileCond"); in TEST_F() local
270 HloInstruction* cond_param = cond_builder.AddInstruction( in TEST_F()
272 HloInstruction* cond_iter = cond_builder.AddInstruction( in TEST_F()
274 HloInstruction* cond_limit = cond_builder.AddInstruction( in TEST_F()
276 HloInstruction* cond_lt = cond_builder.AddInstruction( in TEST_F()
280 module_->AddEmbeddedComputation(cond_builder.Build()); in TEST_F()
|
| D | heap_simulator_test.cc | 48 auto cond_builder = HloComputation::Builder("WhileCond"); in TEST_F() local
50 HloInstruction* cond_param = cond_builder.AddInstruction( in TEST_F()
52 HloInstruction* cond_iter = cond_builder.AddInstruction( in TEST_F()
54 HloInstruction* cond_data = cond_builder.AddInstruction( in TEST_F()
57 HloInstruction* cond_lt = cond_builder.AddInstruction( in TEST_F()
61 module->AddEmbeddedComputation(cond_builder.Build()); in TEST_F()
138 auto cond_builder = HloComputation::Builder("WhileCond"); in TEST_F() local
139 HloInstruction* cond_param = cond_builder.AddInstruction( in TEST_F()
142 cond_builder.AddInstruction(HloInstruction::CreateSlice( in TEST_F()
145 cond_builder.AddInstruction(HloInstruction::CreateReshape(r0f32, slice)); in TEST_F()
[all …]
|
| D | copy_insertion_test.cc | 1524 auto cond_builder = HloComputation::Builder("condition"); in TEST_F() local
1525 cond_builder.AddInstruction( in TEST_F()
1527 auto cond_constant = cond_builder.AddInstruction( in TEST_F()
1529 cond_builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
1532 module->AddEmbeddedComputation(cond_builder.Build()); in TEST_F()
1801 auto cond_builder = HloComputation::Builder("condition"); in TEST_F() local
1802 cond_builder.AddInstruction( in TEST_F()
1804 auto cond_constant = cond_builder.AddInstruction( in TEST_F()
1806 cond_builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F()
1809 module->AddEmbeddedComputation(cond_builder.Build()); in TEST_F()
[all …]
|
| D | hlo_dataflow_analysis_test.cc | 407 auto cond_builder = HloComputation::Builder("condition"); in TEST_P() local
408 auto cond_param = cond_builder.AddInstruction( in TEST_P()
410 auto cond_constant = cond_builder.AddInstruction( in TEST_P()
413 module_->AddEmbeddedComputation(cond_builder.Build()); in TEST_P()
509 auto cond_builder = HloComputation::Builder("condition"); in TEST_P() local
510 cond_builder.AddInstruction( in TEST_P()
512 cond_builder.AddInstruction( in TEST_P()
515 module_->AddEmbeddedComputation(cond_builder.Build()); in TEST_P()
568 auto cond_builder = HloComputation::Builder("condition"); in TEST_P() local
569 cond_builder.AddInstruction( in TEST_P()
[all …]
|
| D | hlo_dce_test.cc | 258 auto cond_builder = HloComputation::Builder(TestName() + "-cond"); in TEST_F() local
260 auto param = cond_builder.AddInstruction( in TEST_F()
262 auto constant = cond_builder.AddInstruction( in TEST_F()
264 cond_builder.AddInstruction( in TEST_F()
268 auto cond_computation = module->AddEmbeddedComputation(cond_builder.Build()); in TEST_F()
|
| D | memory_space_assignment_test.cc | 744 auto cond_builder = HloComputation::Builder("WhileCond"); in TEST_P() local
746 HloInstruction* cond_param = cond_builder.AddInstruction( in TEST_P()
748 HloInstruction* cond_iter = cond_builder.AddInstruction( in TEST_P()
750 HloInstruction* cond_limit = cond_builder.AddInstruction( in TEST_P()
753 HloInstruction* cond_lt = cond_builder.AddInstruction( in TEST_P()
757 module->AddEmbeddedComputation(cond_builder.Build()); in TEST_P()
2627 auto cond_builder = HloComputation::Builder("WhileCond"); in TEST_P() local
2629 HloInstruction* cond_param = cond_builder.AddInstruction( in TEST_P()
2631 HloInstruction* cond_iter = cond_builder.AddInstruction( in TEST_P()
2633 HloInstruction* cond_limit = cond_builder.AddInstruction( in TEST_P()
[all …]
|
| D | dynamic_dimension_inference_test.cc | 755 auto cond_builder = HloComputation::Builder("condition"); in TEST_F() local
756 cond_builder.AddInstruction( in TEST_F()
758 cond_builder.AddInstruction( in TEST_F()
761 module_->AddEmbeddedComputation(cond_builder.Build()); in TEST_F()
|
| D | buffer_assignment_test.cc | 782 auto cond_builder = HloComputation::Builder("WhileCond"); in TEST_F() local
783 HloInstruction* cond_param = cond_builder.AddInstruction( in TEST_F()
785 HloInstruction* cond_iter = cond_builder.AddInstruction( in TEST_F()
787 HloInstruction* cond_limit = cond_builder.AddInstruction( in TEST_F()
789 cond_builder.AddInstruction( in TEST_F()
793 module->AddEmbeddedComputation(cond_builder.Build()); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | outfeed_in_nested_computation_test.cc | 43 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/core/tpu/graph_rewrite/ |
| D | BUILD | 128 ":cond_builder",
189 name = "cond_builder",
190 srcs = ["cond_builder.cc"],
191 hdrs = ["cond_builder.h"],
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | image_ops.cc | 341 xla::XlaBuilder* cond_builder) const { in operator ()()
344 xla::Lt(row_idx, xla::ConstantR0<int32>(cond_builder, num_boxes)); in operator ()()
347 num_outputs_so_far, xla::ConstantR0<int32>(cond_builder, output_size)); in operator ()()
|
| /external/swiftshader/third_party/SPIRV-Tools/source/opt/ |
| D | loop_peeling.cpp | 498 InstructionBuilder cond_builder( in PeelAfter() local
503 return cond_builder in PeelAfter()
504 .AddLessThan(cond_builder in PeelAfter()
|
| /external/deqp-deps/SPIRV-Tools/source/opt/ |
| D | loop_peeling.cpp | 498 InstructionBuilder cond_builder( in PeelAfter() local
503 return cond_builder in PeelAfter()
504 .AddLessThan(cond_builder in PeelAfter()
|
| /external/angle/third_party/vulkan-deps/spirv-tools/src/source/opt/ |
| D | loop_peeling.cpp | 498 InstructionBuilder cond_builder( in PeelAfter() local
503 return cond_builder in PeelAfter()
504 .AddLessThan(cond_builder in PeelAfter()
|