| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| H A D | unfold_large_splat_constant.cc | 54 mlir::OpBuilder op_builder(&module.getBodyRegion()); in MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID() local
61 void MaybeUnfoldLargeSplatConstant(mlir::OpBuilder* op_builder, in MaybeUnfoldLargeSplatConstant()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/kernel/ |
| H A D | tfrt_op_kernel.cc | 241 void TFRTOpMetaMap::RegisterOpMeta(const TFRTOpMetaBuilder& op_builder) { in RegisterOpMeta()
255 TFRTOpRegisterer::TFRTOpRegisterer(const TFRTOpMetaBuilder& op_builder) { in TFRTOpRegisterer()
|
| /aosp_15_r20/external/tensorflow/tensorflow/c/kernels/ops/ |
| H A D | histogram_summary.cc | 30 TF_OpDefinitionBuilder* op_builder = in Register_HistogramSummaryOp() local
|
| H A D | merge_summary.cc | 32 TF_OpDefinitionBuilder* op_builder = in Register_MergeSummaryOp() local
|
| H A D | summary.cc | 33 TF_OpDefinitionBuilder* op_builder = in Register_ScalarSummaryOp() local
|
| H A D | bitcast.cc | 105 TF_OpDefinitionBuilder* op_builder = TF_NewOpDefinitionBuilder("Bitcast"); in RegisterBitcastOp() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/ |
| H A D | op_to_device_cluster.cc | 111 mlir::OpBuilder op_builder(&context); in runOnOperation() local
|
| H A D | tpu_add_resource_device_attribute.cc | 91 mlir::OpBuilder op_builder(&context); in runOnOperation() local
|
| H A D | cluster_function_conversion.cc | 176 mlir::OpBuilder op_builder(&context); in runOnOperation() local
|
| H A D | tpu_integration.cc | 138 mlir::OpBuilder op_builder(&context); in runOnOperation() local
|
| H A D | lower_send_recv.cc | 163 mlir::OpBuilder op_builder(&context); in runOnOperation() local
|
| H A D | merge_clusters.cc | 594 mlir::OpBuilder op_builder(&context); in runOnOperation() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| H A D | mlir_emitter.cc | 119 mlir::OpBuilder op_builder(&function.getBody()); in EmitMlirFuncAndCall() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/ |
| H A D | execution_metadata_exporter.cc | 123 OpMetadataBuilder op_builder(*builder); in CreateSubgraphMetadata() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfr/integration/ |
| H A D | tfr_decompose_ctx.cc | 161 mlir::OpBuilder op_builder(func.getBody()); in ExpandNode() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/ |
| H A D | hexagon_delegate_kernel.cc | 276 auto* op_builder = in BuildGraph() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/ |
| H A D | flatbuffer_import.cc | 1239 OpBuilder op_builder{body}; in ConvertSubgraph() local
1410 OpBuilder op_builder{region}; in AddCallOpInWhileOpRegion() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/task/ |
| H A D | serialization_base.cc | 704 data::GPUOperationBuilder op_builder(*builder); in Encode() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/ |
| H A D | scoped_allocator_optimizer.cc | 666 NodeDefBuilder op_builder(sa_op_name, op_name); in BuildReplacementOp() local
|