Home
last modified time | relevance | path

Searched defs:op_builder (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dunfold_large_splat_constant.cc64 mlir::OpBuilder op_builder(&module.body()); in runOnOperation() local
71 void MaybeUnfoldLargeSplatConstant(mlir::OpBuilder* op_builder, in MaybeUnfoldLargeSplatConstant()
/external/tensorflow/tensorflow/c/kernels/ops/
Dmerge_summary.cc32 TF_OpDefinitionBuilder* op_builder = in Register_MergeSummaryOp() local
Dhistogram_summary.cc30 TF_OpDefinitionBuilder* op_builder = in Register_HistogramSummaryOp() local
Dsummary.cc33 TF_OpDefinitionBuilder* op_builder = in Register_ScalarSummaryOp() local
Dbitcast.cc105 TF_OpDefinitionBuilder* op_builder = TF_NewOpDefinitionBuilder("Bitcast"); in RegisterBitcastOp() local
/external/tensorflow/tensorflow/core/runtime_fallback/kernel/
Dtfrt_op_kernel.cc238 void TFRTOpMetaMap::RegisterOpMeta(const TFRTOpMetaBuilder& op_builder) { in RegisterOpMeta()
252 TFRTOpRegisterer::TFRTOpRegisterer(const TFRTOpMetaBuilder& op_builder) { in TFRTOpRegisterer()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dmlir_emitter.cc116 mlir::OpBuilder op_builder(&function.getBody()); in EmitMlirFuncAndCall() local
/external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/
Dexecution_metadata_exporter.cc120 OpMetadataBuilder op_builder(*builder); in CreateSubgraphMetadata() local
/external/tensorflow/tensorflow/compiler/mlir/tfr/integration/
Dtfr_decompose_ctx.cc159 mlir::OpBuilder op_builder(func.getBody()); in ExpandNode() local
/external/tensorflow/tensorflow/lite/delegates/hexagon/
Dhexagon_delegate_kernel.cc272 auto* op_builder = in BuildGraph() local
/external/tensorflow/tensorflow/compiler/mlir/lite/
Dflatbuffer_import.cc1215 OpBuilder op_builder{body}; in ConvertSubgraph() local
1387 OpBuilder op_builder{region}; in AddCallOpInWhileOpRegion() local
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Dserialization.cc864 data::GPUOperationBuilder op_builder(*builder); in Encode() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dscoped_allocator_optimizer.cc666 NodeDefBuilder op_builder(sa_op_name, op_name); in BuildReplacementOp() local