Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
H A Dlegalize_tf_control_flow.cc73 void ReplaceBlockArgumentsWithImplicitOperands( in ReplaceBlockArgumentsWithImplicitOperands() function
151 ReplaceBlockArgumentsWithImplicitOperands(if_op.getOperation(), inputs); in LowerIf()
176 ReplaceBlockArgumentsWithImplicitOperands(case_op.getOperation(), inputs); in LowerCase()
H A Dlegalize_tf_communication.cc650 void ReplaceBlockArgumentsWithImplicitOperands(mlir::Operation* op, in ReplaceBlockArgumentsWithImplicitOperands() function
692 ReplaceBlockArgumentsWithImplicitOperands(region_if.getOperation(), in ProcessRegionIfOp()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/
H A Dhlo_function_importer.h97 void ReplaceBlockArgumentsWithImplicitOperands(
H A Dhlo_function_importer.cc141 void HloFunctionImporter::ReplaceBlockArgumentsWithImplicitOperands( in ReplaceBlockArgumentsWithImplicitOperands() function in xla::HloFunctionImporter
926 ReplaceBlockArgumentsWithImplicitOperands(op.getOperation(), in ImportInstructionImpl()
957 ReplaceBlockArgumentsWithImplicitOperands(op.getOperation(), in ImportInstructionImpl()