Home
last modified time | relevance | path

Searched refs:defining_tuple_op (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/ir/
H A Dmlir_hlo_builder.cc363 auto defining_tuple_op = result.getDefiningOp<mlir::mhlo::TupleOp>(); in WhileInternal() local
364 return MakeXlaOp(defining_tuple_op); in WhileInternal()
477 auto defining_tuple_op = result.getDefiningOp<mlir::mhlo::TupleOp>(); in RngBitGeneratorInternal() local
478 return MakeXlaOp(defining_tuple_op); in RngBitGeneratorInternal()
616 auto defining_tuple_op = result.getDefiningOp<mlir::mhlo::TupleOp>(); in InfeedWithTokenInternal() local
617 return MakeXlaOp(defining_tuple_op); in InfeedWithTokenInternal()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/
H A Dhlo_function_importer.cc168 auto defining_tuple_op = result.getDefiningOp<mlir::mhlo::TupleOp>(); in CreateTupleFromOpResults() local
169 assert(defining_tuple_op && "builder didn't return the right type"); in CreateTupleFromOpResults()
170 auto tupleOp = defining_tuple_op.getOperation(); in CreateTupleFromOpResults()