| /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/lite/delegates/hexagon/builders/ |
| D | BUILD | 9 name = "op_builder",
25 "op_builder.cc",
58 "op_builder.h",
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/ |
| D | serialization.cc | 877 data::GPUOperationBuilder op_builder(*builder); in Encode() local
878 op_builder.add_arguments(args_fb); in Encode()
879 op_builder.add_code(code_fb); in Encode()
880 op_builder.add_work_group_size(work_group_size_fb); in Encode()
881 op_builder.add_compiler_options(compiler_options_fb_vec); in Encode()
882 op_builder.add_tensor_to_grid(ToFB(op.tensor_to_grid_)); in Encode()
883 op_builder.add_elementwise(op.elementwise_); in Encode()
884 op_builder.add_linkable(op.linkable_); in Encode()
885 op_builder.add_check_src_channels_size(op.check_src_channels_size_); in Encode()
886 op_builder.add_definition(def_fb); in Encode()
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ |
| D | flatbuffer_import.cc | 1007 OpBuilder op_builder{body}; in ConvertSubgraph() local
1027 ConvertMinMaxToStatsOp(tensor, op_builder, input_value)) { in ConvertSubgraph()
1070 op_builder in ConvertSubgraph()
1081 op_builder, const_loc) in ConvertSubgraph()
1083 const_tensor.is_variable, op_builder, const_loc); in ConvertSubgraph()
1115 op_codes, func_names, subgraph.tensors, op_loc, op_builder)); in ConvertSubgraph()
1125 ConvertMinMaxToStatsOp(tensor, op_builder, pair.value())) { in ConvertSubgraph()
1142 op_builder, const_loc) in ConvertSubgraph()
1144 const_tensor.is_variable, op_builder, const_loc); in ConvertSubgraph()
1154 op_builder.create<mlir::ReturnOp>(base_loc, return_operands); in ConvertSubgraph()
[all …]
|
| /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",
81 "//tensorflow/lite/delegates/coreml/builders:op_builder",
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/integration/ |
| D | tfr_decompose_ctx.cc | 159 mlir::OpBuilder op_builder(func.getBody()); in ExpandNode() local
167 mlir::Operation* tf_op = op_builder.createOperation(op_state); in ExpandNode()
168 op_builder.create<mlir::ReturnOp>(loc, tf_op->getResults()); in ExpandNode()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | mlir_emitter.cc | 116 mlir::OpBuilder op_builder(&function.getBody()); in EmitMlirFuncAndCall() local
117 emitter(&op_builder, function); in EmitMlirFuncAndCall()
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/ |
| D | hexagon_delegate_kernel.cc | 272 auto* op_builder = in BuildGraph() local
275 op_builder->PopulateSubGraph(node->inputs, node->outputs, context)); in BuildGraph()
276 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",
|