Searched refs:HloFunctionImporter (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | hlo_function_importer.cc | 108 Status HloFunctionImporter::ImportAsFunc( in ImportAsFunc()
112 HloFunctionImporter importer(module, function_map, builder); in ImportAsFunc()
116 Status HloFunctionImporter::ImportAsRegion( in ImportAsRegion()
119 HloFunctionImporter importer(region->getParentOfType<mlir::ModuleOp>(), {}, in ImportAsRegion()
124 StatusOr<mlir::FuncOp> HloFunctionImporter::ImportAsFunc( in ImportAsFunc()
156 tensorflow::Status HloFunctionImporter::ImportAsRegion( in ImportAsRegion()
169 StatusOr<Value> HloFunctionImporter::ImportInstructionsImpl( in ImportInstructionsImpl()
195 Status HloFunctionImporter::ImportInstructions( in ImportInstructions()
214 StatusOr<Value> HloFunctionImporter::ImportInstructions( in ImportInstructions()
222 HloFunctionImporter importer( in ImportInstructions()
[all …]
|
| D | hlo_module_importer.cc | 46 return HloFunctionImporter::ImportAsFunc( in Import()
50 TF_RETURN_IF_ERROR(HloFunctionImporter::ImportAsFunc( in Import()
|
| D | hlo_function_importer.h | 44 class HloFunctionImporter {
80 HloFunctionImporter(mlir::ModuleOp module, in HloFunctionImporter() function
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | mhlo_to_lhlo_with_xla.cc | 422 TF_RETURN_IF_ERROR(xla::HloFunctionImporter::ImportAsRegion( in EmitSortOp()
497 xla::HloFunctionImporter::ImportInstructions( in EmitFusionOp()
575 TF_RETURN_IF_ERROR(xla::HloFunctionImporter::ImportAsRegion( in EmitScatterOp()
606 TF_RETURN_IF_ERROR(xla::HloFunctionImporter::ImportAsRegion( in EmitSelectAndScatterOp()
609 TF_RETURN_IF_ERROR(xla::HloFunctionImporter::ImportAsRegion( in EmitSelectAndScatterOp()
953 TF_RETURN_IF_ERROR(xla::HloFunctionImporter::ImportAsRegion( in EmitReduceOp()
965 TF_RETURN_IF_ERROR(xla::HloFunctionImporter::ImportAsRegion( in EmitMapOp()
1038 auto replica_groups_attr = xla::HloFunctionImporter::ConvertReplicaGroups( in SetupCommonCollectiveOpAttributes()
1084 TF_RETURN_IF_ERROR(xla::HloFunctionImporter::ImportAsRegion( in EmitAllReduceOp()
1097 xla::HloFunctionImporter::ConvertSourceTargetPairs( in EmitCollectivePermuteOp()
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
| D | mlir_hlo_builder.cc | 592 return HloFunctionImporter::ImportAsRegion(*hlo_module->entry_computation(), in ImportComputation()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 650 HloFunctionImporter::SetLayoutForMlir(load, in EmitUsingElementalIrEmitter()
659 HloFunctionImporter::SetLayoutForMlir( in EmitUsingElementalIrEmitter()
663 HloFunctionImporter::SetLayoutForMlir(fused_copy, output_shape); in EmitUsingElementalIrEmitter()
678 HloFunctionImporter::SetLayoutForMlir(fused_reduce, shape); in EmitUsingElementalIrEmitter()
724 HloFunctionImporter::SetLayoutForMlir(new_op, output_shape); in EmitUsingElementalIrEmitter()
|