| /external/tensorflow/tensorflow/c/kernels/ops/ |
| D | histogram_summary.cc | 30 TF_OpDefinitionBuilder* op_builder = in Register_HistogramSummaryOp() local
32 TF_OpDefinitionBuilderAddInput(op_builder, "tag: string"); in Register_HistogramSummaryOp()
33 TF_OpDefinitionBuilderAddInput(op_builder, "values: T"); in Register_HistogramSummaryOp()
34 TF_OpDefinitionBuilderAddOutput(op_builder, "summary: string"); in Register_HistogramSummaryOp()
35 TF_OpDefinitionBuilderAddAttr(op_builder, "T: realnumbertype = DT_FLOAT"); in Register_HistogramSummaryOp()
37 op_builder, &histogram_summary_shape_inference_fn); in Register_HistogramSummaryOp()
39 TF_RegisterOpDefinition(op_builder, status); in Register_HistogramSummaryOp()
|
| D | summary.cc | 33 TF_OpDefinitionBuilder* op_builder = in Register_ScalarSummaryOp() local
35 TF_OpDefinitionBuilderAddInput(op_builder, "tags: string"); in Register_ScalarSummaryOp()
36 TF_OpDefinitionBuilderAddInput(op_builder, "values: T"); in Register_ScalarSummaryOp()
37 TF_OpDefinitionBuilderAddOutput(op_builder, "summary: string"); in Register_ScalarSummaryOp()
38 TF_OpDefinitionBuilderAddAttr(op_builder, "T: realnumbertype"); in Register_ScalarSummaryOp()
40 op_builder, &scalar_summary_shape_inference_fn); in Register_ScalarSummaryOp()
42 TF_RegisterOpDefinition(op_builder, status); in Register_ScalarSummaryOp()
|
| D | merge_summary.cc | 32 TF_OpDefinitionBuilder* op_builder = in Register_MergeSummaryOp() local
34 TF_OpDefinitionBuilderAddInput(op_builder, "inputs: N * string"); in Register_MergeSummaryOp()
35 TF_OpDefinitionBuilderAddOutput(op_builder, "summary: string"); in Register_MergeSummaryOp()
36 TF_OpDefinitionBuilderAddAttr(op_builder, "N: int >= 1"); in Register_MergeSummaryOp()
38 op_builder, &merge_summary_shape_inference_fn); in Register_MergeSummaryOp()
40 TF_RegisterOpDefinition(op_builder, status); in Register_MergeSummaryOp()
|
| D | bitcast.cc | 105 TF_OpDefinitionBuilder* op_builder = TF_NewOpDefinitionBuilder("Bitcast"); in RegisterBitcastOp() local
106 TF_OpDefinitionBuilderAddInput(op_builder, "input: T"); in RegisterBitcastOp()
107 TF_OpDefinitionBuilderAddOutput(op_builder, "output: type"); in RegisterBitcastOp()
109 op_builder, in RegisterBitcastOp()
114 op_builder, in RegisterBitcastOp()
118 TF_OpDefinitionBuilderSetShapeInferenceFunction(op_builder, in RegisterBitcastOp()
121 TF_RegisterOpDefinition(op_builder, status); in RegisterBitcastOp()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | unfold_large_splat_constant.cc | 54 mlir::OpBuilder op_builder(&module.getBodyRegion()); in MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID() local
56 MaybeUnfoldLargeSplatConstant(&op_builder, const_op); in MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID()
61 void MaybeUnfoldLargeSplatConstant(mlir::OpBuilder* op_builder, in MaybeUnfoldLargeSplatConstant() argument
79 op_builder->setInsertionPoint(const_op); in MaybeUnfoldLargeSplatConstant()
81 op_builder->create<mlir::arith::ConstantOp>( in MaybeUnfoldLargeSplatConstant()
85 op_builder->getI64Type()), in MaybeUnfoldLargeSplatConstant()
88 op_builder->create<mlir::arith::ConstantOp>( in MaybeUnfoldLargeSplatConstant()
94 TFL::FillOp fill = op_builder->create<TFL::FillOp>( in MaybeUnfoldLargeSplatConstant()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/ |
| D | execution_metadata_exporter.cc | 123 OpMetadataBuilder op_builder(*builder); in CreateSubgraphMetadata() local
124 op_builder.add_index(index); in CreateSubgraphMetadata()
126 op_builder.add_hardware(hardware); in CreateSubgraphMetadata()
129 op_builder.add_op_costs(per_device_cost_offset); in CreateSubgraphMetadata()
132 ops.push_back(op_builder.Finish()); in CreateSubgraphMetadata()
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/ |
| D | BUILD | 9 name = "op_builder",
25 "op_builder.cc",
58 "op_builder.h",
|
| /external/tensorflow/tensorflow/core/runtime_fallback/kernel/ |
| D | tfrt_op_kernel.cc | 241 void TFRTOpMetaMap::RegisterOpMeta(const TFRTOpMetaBuilder& op_builder) { in RegisterOpMeta() argument
243 std::make_pair(op_builder.op_name(), op_builder.BuildMeta())); in RegisterOpMeta()
255 TFRTOpRegisterer::TFRTOpRegisterer(const TFRTOpMetaBuilder& op_builder) { in TFRTOpRegisterer() argument
256 tfrt_forwarding_op_meta_map->RegisterOpMeta(op_builder); in TFRTOpRegisterer()
|
| D | tfrt_op_kernel.h | 220 void RegisterOpMeta(const TFRTOpMetaBuilder& op_builder);
237 const TFRTOpMetaBuilder& op_builder);
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/task/ |
| D | serialization_base.cc | 704 data::GPUOperationBuilder op_builder(*builder); in Encode() local
705 op_builder.add_arguments(args_fb); in Encode()
706 op_builder.add_work_group_size(work_group_size_fb); in Encode()
707 op_builder.add_tensor_to_grid(ToFB(op.tensor_to_grid_)); in Encode()
708 op_builder.add_flops(op.flops_); in Encode()
709 op_builder.add_definition(def_fb); in Encode()
710 op_builder.add_grid_dimension(op.grid_dimension_); in Encode()
711 op_builder.add_work_group_launch_order(work_group_launch_order_fb); in Encode()
712 op_builder.add_grid_size(grid_size_fb); in Encode()
713 op_builder.add_src_tensors_names(src_names_fb_vec); in Encode()
[all …]
|
| /external/tensorflow/tensorflow/dtensor/mlir/ |
| D | tpu_integration.cc | 138 mlir::OpBuilder op_builder(&context); in runOnOperation() local
148 &op_builder, &cluster))) in runOnOperation()
152 &op_builder); in runOnOperation()
|
| D | op_to_device_cluster.cc | 111 mlir::OpBuilder op_builder(&context); in runOnOperation() local
128 if (mlir::failed(WrapDeviceCluster(&op_builder, operation))) in runOnOperation()
|
| D | cluster_function_conversion.cc | 176 mlir::OpBuilder op_builder(&context); in runOnOperation() local
179 ReplaceClusterWithPartitionCallOp(&op_builder, cluster_func))) { in runOnOperation()
|
| D | tpu_add_resource_device_attribute.cc | 91 mlir::OpBuilder op_builder(&context); in runOnOperation() local
|
| D | lower_send_recv.cc | 163 mlir::OpBuilder op_builder(&context); in runOnOperation() local
|
| D | merge_clusters.cc | 594 mlir::OpBuilder op_builder(&context); in runOnOperation() local
|
| /external/tensorflow/tensorflow/lite/delegates/coreml/ |
| D | coreml_delegate_kernel.mm | 126 auto* op_builder = builder_->AddBuilder(reg->builtin_code, node);
127 if (op_builder == nullptr) {
131 if (op_builder->RegisterInputs(node->inputs, context) != kTfLiteOk) {
135 if (op_builder->PopulateSubgraph(context) != kTfLiteOk) {
139 if (op_builder->RegisterOutputs(node->outputs, context) != kTfLiteOk) {
|
| D | BUILD | 61 "//tensorflow/lite/delegates/coreml/builders:op_builder",
82 "//tensorflow/lite/delegates/coreml/builders:op_builder",
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/integration/ |
| D | tfr_decompose_ctx.cc | 161 mlir::OpBuilder op_builder(func.getBody()); in ExpandNode() local
169 mlir::Operation* tf_op = op_builder.create(op_state); in ExpandNode()
170 op_builder.create<mlir::func::ReturnOp>(loc, tf_op->getResults()); in ExpandNode()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ |
| D | flatbuffer_import.cc | 1239 OpBuilder op_builder{body}; in ConvertSubgraph() local
1259 ConvertMinMaxToStatsOp(tensor, op_builder, input_value)) { in ConvertSubgraph()
1307 op_builder in ConvertSubgraph()
1318 op_builder, const_loc) in ConvertSubgraph()
1320 const_tensor.is_variable, op_builder, const_loc); in ConvertSubgraph()
1348 op_codes, func_names, subgraph.tensors, op_loc, op_builder)); in ConvertSubgraph()
1358 ConvertMinMaxToStatsOp(tensor, op_builder, pair.value())) { in ConvertSubgraph()
1375 op_builder, const_loc) in ConvertSubgraph()
1377 const_tensor.is_variable, op_builder, const_loc); in ConvertSubgraph()
1387 op_builder.create<mlir::func::ReturnOp>(base_loc, return_operands); in ConvertSubgraph()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | mlir_emitter.cc | 119 mlir::OpBuilder op_builder(&function.getBody()); in EmitMlirFuncAndCall() local
120 emitter(&op_builder, function); in EmitMlirFuncAndCall()
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/ |
| D | hexagon_delegate_kernel.cc | 276 auto* op_builder = in BuildGraph() local
279 op_builder->PopulateSubGraph(node->inputs, node->outputs, context)); in BuildGraph()
280 TF_LITE_ENSURE_STATUS(op_builder->RegisterOutputs(node->outputs, context)); in BuildGraph()
|
| D | BUILD | 60 "//tensorflow/lite/delegates/hexagon/builders:op_builder",
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | scoped_allocator_optimizer.cc | 666 NodeDefBuilder op_builder(sa_op_name, op_name); in BuildReplacementOp() local
667 op_builder.Device(device_name); in BuildReplacementOp()
676 op_builder.Attr(it.first, it.second); in BuildReplacementOp()
678 op_builder.Attr("_forward_input", {0, 0}); in BuildReplacementOp()
679 op_builder.Input(sac_name, 0, dtype); in BuildReplacementOp()
681 LOG_WARNING_AND_RETURN_IF_ERROR(op_builder.Finalize(sa_op_node)); in BuildReplacementOp()
|
| /external/tensorflow/tensorflow/lite/delegates/coreml/builders/ |
| D | BUILD | 24 name = "op_builder",
|