Home
last modified time | relevance | path

Searched refs:ImportAsRegion (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/xla/
Dhlo_function_importer.cc116 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 …]
Dhlo_function_importer.h55 static Status ImportAsRegion(const xla::HloComputation& computation,
97 tensorflow::Status ImportAsRegion(const HloComputation& computation,
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dmhlo_to_lhlo_with_xla.cc422 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/
Dmlir_hlo_builder.cc592 return HloFunctionImporter::ImportAsRegion(*hlo_module->entry_computation(), in ImportComputation()