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
42 nested_computation_(nested_computation) {} in IrEmitterNested()
46 const HloComputation& nested_computation, in Create() argument
49 hlo_module_config, nested_computation, ir_emitter_context)); in Create()
50 TF_RETURN_IF_ERROR(emitter->EmitConstants(nested_computation, false)); in Create()
|
| D | ir_emitter_nested.h | 44 const HloComputation& nested_computation,
68 const HloComputation& nested_computation,
|
| D | ir_emitter.h | 152 Status EmitCallToNestedComputation(const HloComputation& nested_computation,
161 const HloComputation& nested_computation, llvm::Value* output_address,
220 const HloComputation& nested_computation, llvm::Type* element_ir_type);
|
| D | ir_emitter.cc | 225 const HloComputation& nested_computation, in EmitCallToNestedComputation() argument
227 TF_RET_CHECK(nested_computation.num_parameters() > 0); in EmitCallToNestedComputation()
229 computation_to_ir_function_[&nested_computation]; in EmitCallToNestedComputation()
233 IrEmitterNested::Create(hlo_module_config_, nested_computation, in EmitCallToNestedComputation()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_module.cc | 571 HloComputation* nested_computation = AddEmbeddedComputation( in OutlineExpressionFromComputation() local
574 output->shape(), arguments, nested_computation)); in OutlineExpressionFromComputation()
581 VLOG(2) << "to " << nested_computation->ToString(); in OutlineExpressionFromComputation()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | cpu_gpu_fusion_test.cc | 680 auto nested_computation = in XLA_TEST_F() local
685 nested_computation)); in XLA_TEST_F()
|