| /external/tensorflow/tensorflow/compiler/xla/service/ |
| flatten_call_graph.cc | 52 if (computation == instruction->true_computation()) {
|
| layout_assignment.cc | 597 HloComputation* true_computation = instruction->true_computation(); local
602 TF_RET_CHECK(true_computation->num_parameters() == 1);
605 FindOrDie(computation_layouts_, true_computation);
712 HloComputation* true_computation = instruction->true_computation(); local
722 true_computation->root_instruction()->shape()));
[all...] |
| call_graph_test.cc | 243 HloComputation* true_computation = local
257 kScalarShape, pred, const1, true_computation, const2,
273 UnorderedElementsAre(true_computation, false_computation));
277 const CallGraphNode& true_node = call_graph->GetNode(true_computation);
|
| shape_inference.h | 196 const Shape& false_operand, const ProgramShape& true_computation,
|
| user_computation.h | 235 const UserComputation& true_computation,
|
| shape_inference.cc | [all...] |
| hlo_ordering.cc | 192 conditional->true_computation())) {
|
| hlo_dataflow_analysis.cc | 368 conditional->true_computation()->root_instruction()),
473 if (parameter->parent() == conditional->true_computation()) {
628 // If operand 1 is the use of instruction, then the true_computation's
636 add_to_worklist(user->true_computation()->parameter_instruction(0));
[all...] |
| hlo_instruction.h | 451 HloInstruction* true_computation_arg, HloComputation* true_computation,
[all...] |
| layout_assignment_test.cc | 689 HloComputation* true_computation = local
705 result_tshape, pred, tuple, true_computation, tuple, false_computation));
712 const HloInstruction* true_root = true_computation->root_instruction();
[all...] |
| hlo_instruction.cc | 484 HloInstruction* true_computation_arg, HloComputation* true_computation,
491 // In called_computations_, the index of true_computation must be 0 and that
494 instruction->called_computations_.push_back(true_computation);
2059 HloComputation* HloInstruction::true_computation() const { function in class:xla::HloInstruction
[all...] |
| user_computation.cc | 790 const UserComputation& true_computation,
802 true_computation.version();
805 true_computation.ComputeProgramShape(true_computation_version));
3292 HloComputation* true_computation = ResolveComputation( local
[all...] |
| service.cc | [all...] |
| buffer_assignment_test.cc | 754 auto true_computation = module->AddEmbeddedComputation( local
[all...] |
| hlo_cost_analysis.cc | 521 ProcessSubcomputation(conditional->true_computation()));
|
| hlo_dataflow_analysis_test.cc | 1563 HloComputation* true_computation = local
1642 HloComputation* true_computation = local
[all...] |
| hlo_module.cc | 151 replacements, instruction->true_computation(), nullptr);
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| local_computation_builder.cc | 510 const LocalComputation& true_computation,
514 true_computation.computation(), false_operand,
|
| local_computation_builder.h | 263 const LocalComputation& true_computation,
|
| xla_client_test.py | [all...] |
| xla_client.py | [all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| ir_emitter_unnested.cc | [all...] |
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| conditional_test.cc | 176 auto true_computation = CreateR0ConstantComputation(56.0f); local
178 auto result = builder.Conditional(pred, operands, true_computation, operands,
571 "only parameter of true_computation"));
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| xla_data.proto | 736 ComputationHandle true_computation = 4;
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| ir_emitter.cc | [all...] |
