Searched refs:nested_computation (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_nested.cc | 39 const HloComputation& nested_computation, in IrEmitterNested() argument
44 EmitBasePointersForNestedComputation(nested_computation, &io_hlos); in IrEmitterNested()
48 const HloComputation& nested_computation, in EmitBasePointersForNestedComputation() argument
53 nested_computation.parameter_instructions()) { in EmitBasePointersForNestedComputation()
63 const HloInstruction* root = nested_computation.root_instruction(); in EmitBasePointersForNestedComputation()
82 nested_computation.name())), // The name of the function. in EmitBasePointersForNestedComputation()
102 for (const auto* hlo : nested_computation.instructions()) { in EmitBasePointersForNestedComputation()
104 hlo != nested_computation.root_instruction()) { in EmitBasePointersForNestedComputation()
|
| D | ir_emitter_nested.h | 46 const HloComputation& nested_computation,
63 const HloComputation& nested_computation,
|
| D | ir_emitter.h | 156 Status EmitCallToNestedComputation(const HloComputation& nested_computation,
165 const HloComputation& nested_computation, llvm::Value* output_address,
233 const HloComputation& nested_computation, llvm::Type* element_ir_type);
|
| D | ir_emitter.cc | 152 const HloComputation& nested_computation, in EmitCallToNestedComputation() argument
154 TF_RET_CHECK(nested_computation.num_parameters() > 0); in EmitCallToNestedComputation()
156 computation_to_ir_function_[&nested_computation]; in EmitCallToNestedComputation()
158 IrEmitterNested ir_emitter_nested(hlo_module_config_, nested_computation, in EmitCallToNestedComputation()
161 nested_computation.root_instruction()->Accept(&ir_emitter_nested)); in EmitCallToNestedComputation()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_module.cc | 470 HloComputation* nested_computation = AddEmbeddedComputation( in OutlineExpressionFromComputation() local
473 output->shape(), arguments, nested_computation)); in OutlineExpressionFromComputation()
480 VLOG(2) << "to " << nested_computation->ToString(); in OutlineExpressionFromComputation()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | fusion_test.cc | 682 auto nested_computation = in XLA_TEST_F() local
687 nested_computation)); in XLA_TEST_F()
|