| /external/tensorflow/tensorflow/lite/kernels/ |
| D | detection_postprocess.cc | 96 auto* op_data = new OpData; in Init() local
99 op_data->max_detections = m["max_detections"].AsInt32(); in Init()
100 op_data->max_classes_per_detection = m["max_classes_per_detection"].AsInt32(); in Init()
102 op_data->detections_per_class = kNumDetectionsPerClass; in Init()
104 op_data->detections_per_class = m["detections_per_class"].AsInt32(); in Init()
106 op_data->use_regular_non_max_suppression = false; in Init()
108 op_data->use_regular_non_max_suppression = m["use_regular_nms"].AsBool(); in Init()
110 op_data->non_max_suppression_score_threshold = in Init()
112 op_data->intersection_over_union_threshold = m["nms_iou_threshold"].AsFloat(); in Init()
113 op_data->num_classes = m["num_classes"].AsInt32(); in Init()
[all …]
|
| D | while.cc | 94 auto* op_data = new OpData; in Init() local
97 op_data->cond_subgraph_index = m["cond_subgraph_index"].AsInt32(); in Init()
98 op_data->body_subgraph_index = m["body_subgraph_index"].AsInt32(); in Init()
99 op_data->cond_has_dynamic_output_tensors = false; in Init()
100 op_data->body_has_dynamic_output_tensors = false; in Init()
101 return op_data; in Init()
109 OpData* op_data = reinterpret_cast<OpData*>(node->user_data); in Prepare() local
117 TF_LITE_ENSURE(context, op_data->cond_subgraph_index < subgraphs->size()); in Prepare()
118 TF_LITE_ENSURE(context, op_data->body_subgraph_index < subgraphs->size()); in Prepare()
120 Subgraph* cond_subgraph = (*subgraphs)[op_data->cond_subgraph_index].get(); in Prepare()
[all …]
|
| D | if.cc | 32 auto* op_data = new OpData; in Init() local
35 op_data->then_subgraph_index = m["then_subgraph_index"].AsInt32(); in Init()
36 op_data->else_subgraph_index = m["else_subgraph_index"].AsInt32(); in Init()
37 return op_data; in Init()
45 const OpData* op_data = reinterpret_cast<OpData*>(node->user_data); in Prepare() local
65 TF_LITE_ENSURE(context, op_data->then_subgraph_index < subgraphs->size()); in Prepare()
66 TF_LITE_ENSURE(context, op_data->else_subgraph_index < subgraphs->size()); in Prepare()
68 Subgraph* then_subgraph = (*subgraphs)[op_data->then_subgraph_index].get(); in Prepare()
69 Subgraph* else_subgraph = (*subgraphs)[op_data->else_subgraph_index].get(); in Prepare()
129 const OpData* op_data = reinterpret_cast<OpData*>(node->user_data); in Eval() local
[all …]
|
| D | lstm.cc | 108 auto* op_data = new OpData(); in Init() local
109 op_data->kernel_type = kTfLiteLSTMFullKernel; in Init()
111 &op_data->scratch_tensor_index); in Init()
112 return op_data; in Init()
303 OpData* op_data = reinterpret_cast<OpData*>(node->user_data); in Prepare() local
317 op_data->is_layer_norm_lstm = false; in Prepare()
319 op_data->is_layer_norm_lstm = true; in Prepare()
323 op_data->is_layer_norm_lstm = false; in Prepare()
331 const bool is_layer_norm_lstm = op_data->is_layer_norm_lstm; in Prepare()
332 op_data->activation_state_tensor_index = in Prepare()
[all …]
|
| D | dequantize.cc | 46 auto* op_data = new OpData(); in Init() local
47 op_data->float_dequantized_weights_initialized = false; in Init()
48 return op_data; in Init()
75 OpData* op_data = reinterpret_cast<OpData*>(node->user_data); in Eval() local
78 op_data->float_dequantized_weights_initialized) { in Eval()
106 op_data->float_dequantized_weights_initialized = true; in Eval()
|
| D | mirror_pad.cc | 50 OpData* op_data = nullptr; member
100 auto& cache_entry = eval_data->op_data->cache[cache_index]; in Eval()
160 OpData* op_data = reinterpret_cast<OpData*>(node->user_data); in Eval() local
190 std::fill(op_data->cache.begin(), op_data->cache.end(), in Eval()
200 eval_data.op_data = op_data; \ in Eval()
243 OpData* op_data = reinterpret_cast<OpData*>(node->user_data); in Prepare() local
250 op_data->cache.resize(num_elements + 1); in Prepare()
|
| D | svdf.cc | 114 auto* op_data = new OpData(); in Init() local
115 op_data->float_weights_time_initialized = false; in Init()
117 &op_data->scratch_tensor_index); in Init()
118 return op_data; in Init()
127 OpData* op_data = reinterpret_cast<OpData*>(node->user_data); in Prepare() local
128 int scratch_tensor_index = op_data->scratch_tensor_index; in Prepare()
133 op_data->activation_state_tensor_index = in Prepare()
162 &context->tensors[op_data->activation_state_tensor_index]; in Prepare()
365 OpData* op_data = reinterpret_cast<OpData*>(node->user_data); in Eval() local
377 &context->tensors[op_data->activation_state_tensor_index]; in Eval()
[all …]
|
| D | reduce.cc | 65 auto* op_data = new OpData(); in Init() local
66 context->AddTensors(context, 3, &op_data->scratch_tensor_index); in Init()
67 return op_data; in Init()
163 OpData* op_data = reinterpret_cast<OpData*>(node->user_data); in InitializeTemporaries() local
166 node->temporaries->data[0] = op_data->scratch_tensor_index; in InitializeTemporaries()
176 node->temporaries->data[1] = op_data->scratch_tensor_index + 1; in InitializeTemporaries()
180 node->temporaries->data[2] = op_data->scratch_tensor_index + 2; in InitializeTemporaries()
|
| D | fully_connected.cc | 72 auto* op_data = new OpData(); in Init() local
74 &op_data->scratch_tensor_index); in Init()
75 return op_data; in Init()
|
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | kernel.cc | 370 auto* op_data = new OpData; in Init() local
377 op_data->eager_context = in Init()
380 op_data->buffer_map = reinterpret_cast<DelegateData*>(params->delegate->data_) in Init()
386 op_data->subgraph_outputs.push_back(tensor_index); in Init()
392 op_data->subgraph_inputs.push_back(tensor_index); in Init()
395 op_data->nodes.reserve(params->nodes_to_replace->size); in Init()
404 op_data->nodes.emplace_back(new OpNode(node->inputs, node->outputs)); in Init()
405 OpNode& node_data = *op_data->nodes.back(); in Init()
413 status = node_data.BuildEagerOp(op_data->eager_context); in Init()
421 return op_data; in Init()
[all …]
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | shape_function_test.cc | 26 std::vector<OpRegistrationData> op_data; in TEST() local
27 op_registry->GetOpRegistrationData(&op_data); in TEST()
28 for (const OpRegistrationData& op_reg_data : op_data) { in TEST()
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | identify_relu1.cc | 46 const auto& op_data = op_array.GetBuffer<ArrayDataType::kFloat>().data; in CheckArrayIsScalarFloat() local
47 return op_data[0] == val; in CheckArrayIsScalarFloat()
|
| /external/ltp/utils/ffsb-6.0-rc2/ |
| D | ffsb_fs.c | 80 memcpy(target->op_data, orig->op_data, sizeof(void *) * FFSB_NUMOPS); in clone_ffsb_fs()
544 fs->op_data[opnum] = data; in fs_set_opdata()
549 return fs->op_data[opnum]; in fs_get_opdata()
|
| D | ffsb_fs.h | 107 void *op_data[FFSB_NUMOPS]; member
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | op.cc | 134 std::vector<OpRegistrationData>* op_data) { in GetOpRegistrationData() argument
138 op_data->push_back(*p.second); in GetOpRegistrationData()
|
| D | op.h | 93 void GetOpRegistrationData(std::vector<OpRegistrationData>* op_data);
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties.cc | 563 const OpRegistrationData* op_data; member
868 if (ctx->op_data && ctx->op_data->is_function_op) { in UpdateNode()
1072 TF_RETURN_IF_ERROR(function_library_.LookUp(node->op(), &node_ctx.op_data)); in AddNode()
1074 if (node_ctx.op_data->is_function_op) { in AddNode()
1078 TF_RETURN_IF_ERROR(InOutTypesForNode(*node, node_ctx.op_data->op_def, in AddNode()
1091 graph_def_version_, node, node_ctx.op_data->op_def, input_shapes, in AddNode()
1184 if (!IsWhiteListedOpTypeForEvaluateNode(c->op_data->op_def.name())) { in ShouldUpdateOutputShapesAndValues()
1582 if (!c->op_data || c->op_data->shape_inference_fn == nullptr) { in InferShapes()
1589 c->inference_context->Run(c->op_data->shape_inference_fn)); in InferShapes()
|
| /external/tensorflow/tensorflow/lite/g3doc/guide/ |
| D | ops_custom.md | 151 auto* op_data = reinterpret_cast<OpData*>(node->user_data);
154 node->temporaries->data[0] = op_data->temp_tensor_index;
155 TfLiteTensor* temp_tensor = &context->tensors[op_data->temp_tensor_index];
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | context.cc | 236 const string& name, const tensorflow::OpRegistrationData** op_data) { in FindFunctionOpData() argument
238 return func_lib_def_.LookUp(name, op_data); in FindFunctionOpData()
|
| D | context.h | 137 const tensorflow::OpRegistrationData** op_data);
|