Searched refs:nested_computation (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | ir_emitter_nested.cc | 44 const HloComputation& nested_computation, in IrEmitterNested() argument
47 nested_computation_(nested_computation) {} in IrEmitterNested()
51 const HloComputation& nested_computation, in Create() argument
54 hlo_module_config, nested_computation, ir_emitter_context)); in Create()
55 TF_RETURN_IF_ERROR(emitter->EmitConstants(nested_computation)); in Create()
|
| H A D | ir_emitter.h | 141 Status EmitCallToNestedComputation(const HloComputation& nested_computation,
150 const HloComputation& nested_computation, llvm::Value* output_address,
215 const HloComputation& nested_computation, llvm::Type* element_ir_type);
|
| H A D | ir_emitter_nested.h | 53 const HloComputation& nested_computation,
78 const HloComputation& nested_computation,
|
| H A D | ir_emitter.cc | 154 const HloComputation& nested_computation, in EmitCallToNestedComputation() argument
156 TF_RET_CHECK(nested_computation.num_parameters() > 0); in EmitCallToNestedComputation()
158 computation_to_ir_function_[&nested_computation]; in EmitCallToNestedComputation()
162 IrEmitterNested::Create(hlo_module_config_, nested_computation, in EmitCallToNestedComputation()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | hlo_module.cc | 685 HloComputation* nested_computation = AddEmbeddedComputation( in OutlineExpressionFromComputation() local
688 output->shape(), arguments, nested_computation)); in OutlineExpressionFromComputation()
695 VLOG(2) << "to " << nested_computation->ToString(); in OutlineExpressionFromComputation()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | cpu_gpu_fusion_test.cc | 678 auto nested_computation = in XLA_TEST_F() local
683 nested_computation)); in XLA_TEST_F()
|