Home
last modified time | relevance | path

Searched refs:nested_computation (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_nested.cc38 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()
Dir_emitter_nested.h46 const HloComputation& nested_computation,
63 const HloComputation& nested_computation,
Dir_emitter.h142 const HloComputation& nested_computation,
150 const HloComputation& nested_computation, llvm::Value* output_address,
208 const HloComputation& nested_computation, llvm::Type* element_ir_type);
Dir_emitter.cc157 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/
Dhlo_module.cc458 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/
Dfusion_test.cc657 auto nested_computation = in XLA_TEST_F() local
662 nested_computation)); in XLA_TEST_F()