Searched defs:op_builder (Results 1 – 13 of 13) sorted by relevance
64 mlir::OpBuilder op_builder(&module.body()); in runOnOperation() local71 void MaybeUnfoldLargeSplatConstant(mlir::OpBuilder* op_builder, in MaybeUnfoldLargeSplatConstant()
32 TF_OpDefinitionBuilder* op_builder = in Register_MergeSummaryOp() local
30 TF_OpDefinitionBuilder* op_builder = in Register_HistogramSummaryOp() local
33 TF_OpDefinitionBuilder* op_builder = in Register_ScalarSummaryOp() local
105 TF_OpDefinitionBuilder* op_builder = TF_NewOpDefinitionBuilder("Bitcast"); in RegisterBitcastOp() local
238 void TFRTOpMetaMap::RegisterOpMeta(const TFRTOpMetaBuilder& op_builder) { in RegisterOpMeta()252 TFRTOpRegisterer::TFRTOpRegisterer(const TFRTOpMetaBuilder& op_builder) { in TFRTOpRegisterer()
116 mlir::OpBuilder op_builder(&function.getBody()); in EmitMlirFuncAndCall() local
120 OpMetadataBuilder op_builder(*builder); in CreateSubgraphMetadata() local
159 mlir::OpBuilder op_builder(func.getBody()); in ExpandNode() local
272 auto* op_builder = in BuildGraph() local
1215 OpBuilder op_builder{body}; in ConvertSubgraph() local1387 OpBuilder op_builder{region}; in AddCallOpInWhileOpRegion() local
864 data::GPUOperationBuilder op_builder(*builder); in Encode() local
666 NodeDefBuilder op_builder(sa_op_name, op_name); in BuildReplacementOp() local