Searched refs:nested_computation (Results 1 – 6 of 6) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | ir_emitter_nested.cc | 38 const HloComputation& nested_computation, in IrEmitterNested() argument 43 EmitBasePointersForNestedComputation(nested_computation, &io_hlos); in IrEmitterNested() 47 const HloComputation& nested_computation, in EmitBasePointersForNestedComputation() argument 52 nested_computation.parameter_instructions()) { in EmitBasePointersForNestedComputation() 62 const HloInstruction* root = nested_computation.root_instruction(); in EmitBasePointersForNestedComputation() 81 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 | 142 const HloComputation& nested_computation, 150 const HloComputation& nested_computation, llvm::Value* output_address, 208 const HloComputation& nested_computation, llvm::Type* element_ir_type);
|
D | ir_emitter.cc | 157 const HloComputation& nested_computation, in EmitCallToNestedComputation() argument 159 TF_RET_CHECK(nested_computation.num_parameters() > 0); in EmitCallToNestedComputation() 161 computation_to_ir_function_[&nested_computation]; in EmitCallToNestedComputation() 163 IrEmitterNested ir_emitter_nested(hlo_module_config_, nested_computation, in EmitCallToNestedComputation() 166 nested_computation.root_instruction()->Accept(&ir_emitter_nested)); in EmitCallToNestedComputation()
|
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | hlo_module.cc | 458 HloComputation* nested_computation = AddEmbeddedComputation( in OutlineExpressionFromComputation() local 461 output->shape(), arguments, nested_computation)); in OutlineExpressionFromComputation() 468 VLOG(2) << "to " << nested_computation->ToString(); in OutlineExpressionFromComputation()
|
/external/tensorflow/tensorflow/compiler/xla/tests/ |
D | fusion_test.cc | 657 auto nested_computation = in XLA_TEST_F() local 662 nested_computation)); in XLA_TEST_F()
|