Searched refs:ImportAsRegion (Results 1 – 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/compiler/mlir/xla/ |
D | hlo_function_importer.cc | 116 Status HloFunctionImporter::ImportAsRegion( in ImportAsRegion() function in xla::HloFunctionImporter 121 return importer.ImportAsRegion(computation, region); in ImportAsRegion() 156 tensorflow::Status HloFunctionImporter::ImportAsRegion( in ImportAsRegion() function in xla::HloFunctionImporter 444 TF_RETURN_IF_ERROR(ImportAsRegion(*scatter->to_apply(), in ImportInstructionImpl() 466 TF_RETURN_IF_ERROR(ImportAsRegion(*select_scatter->select(), in ImportInstructionImpl() 468 TF_RETURN_IF_ERROR(ImportAsRegion(*select_scatter->scatter(), in ImportInstructionImpl() 499 ImportAsRegion(*sort_instruction->to_apply(), &sort_op.comparator())); in ImportInstructionImpl() 522 TF_RETURN_IF_ERROR(ImportAsRegion(*instruction->true_computation(), in ImportInstructionImpl() 524 TF_RETURN_IF_ERROR(ImportAsRegion(*instruction->false_computation(), in ImportInstructionImpl() 541 TF_RETURN_IF_ERROR(ImportAsRegion(*computation, &op.branches()[index])); in ImportInstructionImpl() [all …]
|
D | hlo_function_importer.h | 55 static Status ImportAsRegion(const xla::HloComputation& computation, 97 tensorflow::Status ImportAsRegion(const HloComputation& computation,
|
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
D | mhlo_to_lhlo_with_xla.cc | 422 TF_RETURN_IF_ERROR(xla::HloFunctionImporter::ImportAsRegion( in EmitSortOp() 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() 1084 TF_RETURN_IF_ERROR(xla::HloFunctionImporter::ImportAsRegion( in EmitAllReduceOp() 1224 TF_RETURN_IF_ERROR(xla::HloFunctionImporter::ImportAsRegion( in EmitReduceWindowOp()
|
/external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
D | mlir_hlo_builder.cc | 592 return HloFunctionImporter::ImportAsRegion(*hlo_module->entry_computation(), in ImportComputation()
|