Home
last modified time | relevance | path

Searched refs:op_data (Results 1 – 25 of 35) sorted by relevance

12

/external/tensorflow/tensorflow/lite/experimental/acceleration/mini_benchmark/
Ddecode_jpeg.cc52 OpData* op_data = new OpData(); in Init() local
53 op_data->height = m["height"].AsInt32(); in Init()
54 op_data->width = m["width"].AsInt32(); in Init()
55 op_data->num_images = m["num_images"].AsInt32(); in Init()
56 op_data->channels = m["channels"].AsInt32(); in Init()
57 return op_data; in Init()
66 OpData* op_data = reinterpret_cast<OpData*>(node->user_data); in Prepare() local
67 TF_LITE_ENSURE(context, op_data); in Prepare()
68 TF_LITE_ENSURE(context, op_data->height > 0); in Prepare()
69 TF_LITE_ENSURE(context, op_data->width > 0); in Prepare()
[all …]
Dcall.cc179 auto* op_data = new OpData; in Init() local
185 op_data->subgraph_index = map["subgraph_index"].AsInt32(); in Init()
186 op_data->loop_count = map["loop_count"].AsInt32(); in Init()
187 return op_data; in Init()
195 OpData* op_data = reinterpret_cast<OpData*>(node->user_data); in Prepare() local
196 TF_LITE_ENSURE(context, op_data); in Prepare()
202 (op_data->subgraph_index < subgraphs->size()) && in Prepare()
203 (op_data->subgraph_index >= 0), in Prepare()
205 Subgraph* subgraph = (*subgraphs)[op_data->subgraph_index].get(); in Prepare()
209 int loop_count = op_data->loop_count; in Prepare()
[all …]
/external/tensorflow/tensorflow/lite/kernels/perception/
Dmax_pool_with_argmax.cc120 OpData* op_data = new OpData; in Init() local
121 op_data->params.computed.padding = TfLitePaddingValues{0, 0, 0, 0}; in Init()
122 op_data->include_batch_in_index = m[kIncludeBatchStr].AsBool(); in Init()
123 op_data->params.activation = kTfLiteActNone; in Init()
127 op_data->params.padding = kTfLitePaddingValid; in Init()
129 op_data->params.padding = kTfLitePaddingSame; in Init()
131 op_data->params.padding = kTfLitePaddingUnknown; in Init()
139 op_data->params.filter_height = pool_size[1].AsInt32(); in Init()
140 op_data->params.filter_width = pool_size[2].AsInt32(); in Init()
147 op_data->params.stride_height = strides[1].AsInt32(); in Init()
[all …]
/external/tensorflow/tensorflow/lite/kernels/
Dwhile.cc118 const OpData* op_data = reinterpret_cast<OpData*>(node->user_data); in DeepCopyTensorsShapeTypeData() local
120 if (op_data->body_has_dynamic_output_tensors) { in DeepCopyTensorsShapeTypeData()
144 const OpData* op_data = reinterpret_cast<OpData*>(node->user_data); in ShallowCopyTensorsShapeTypeData() local
146 TF_LITE_ENSURE_EQ(context, op_data->body_has_dynamic_output_tensors, true); in ShallowCopyTensorsShapeTypeData()
207 auto* op_data = new OpData; in Init() local
209 op_data->cond_subgraph_index = params->cond_subgraph_index; in Init()
210 op_data->body_subgraph_index = params->body_subgraph_index; in Init()
211 op_data->cond_has_dynamic_output_tensors = false; in Init()
212 op_data->body_has_dynamic_output_tensors = false; in Init()
213 op_data->body_use_shallow_copy = false; in Init()
[all …]
Ddetection_postprocess.cc104 auto* op_data = new OpData; in Init() local
107 op_data->max_detections = m["max_detections"].AsInt32(); in Init()
108 op_data->max_classes_per_detection = m["max_classes_per_detection"].AsInt32(); in Init()
110 op_data->detections_per_class = kNumDetectionsPerClass; in Init()
112 op_data->detections_per_class = m["detections_per_class"].AsInt32(); in Init()
114 op_data->use_regular_non_max_suppression = false; in Init()
116 op_data->use_regular_non_max_suppression = m["use_regular_nms"].AsBool(); in Init()
118 op_data->non_max_suppression_score_threshold = in Init()
120 op_data->intersection_over_union_threshold = m["nms_iou_threshold"].AsFloat(); in Init()
121 op_data->num_classes = m["num_classes"].AsInt32(); in Init()
[all …]
Dcall_once.cc46 auto* op_data = new OpData; in Init() local
48 op_data->init_subgraph_index = params->init_subgraph_index; in Init()
49 return op_data; in Init()
57 const OpData* op_data = reinterpret_cast<OpData*>(node->user_data); in Prepare() local
65 resource::GetInitializationStatus(map, op_data->init_subgraph_index); in Prepare()
73 TF_LITE_ENSURE(context, op_data->init_subgraph_index < subgraphs->size()); in Prepare()
76 Subgraph* init_subgraph = (*subgraphs)[op_data->init_subgraph_index].get(); in Prepare()
83 OpData* op_data = reinterpret_cast<OpData*>(node->user_data); in Eval() local
91 resource::GetInitializationStatus(map, op_data->init_subgraph_index); in Eval()
95 Subgraph& init_subgraph = *(*subgraphs)[op_data->init_subgraph_index]; in Eval()
Dnumeric_verify.cc69 auto* op_data = new OpData(); in Init() local
70 op_data->float_input_initialized = false; in Init()
76 op_data->tolerance = tolerance; in Init()
77 op_data->log_if_failed = log_if_failed; in Init()
79 return op_data; in Init()
89 OpData* op_data = reinterpret_cast<OpData*>(node->user_data); in Prepare() local
100 if (op_data->cache_tensor_id == kTensorNotAllocated) { in Prepare()
102 context, context->AddTensors(context, 1, &op_data->cache_tensor_id)); in Prepare()
107 node->temporaries->data[0] = op_data->cache_tensor_id; in Prepare()
143 OpData* op_data = reinterpret_cast<OpData*>(node->user_data); in Eval() local
[all …]
Delementwise.cc98 auto* op_data = static_cast<OpData*>(node->user_data); in GenericPrepare() local
115 op_data->input_offset = input_params->zero_point->data[0]; in GenericPrepare()
116 op_data->output_offset = output_params->zero_point->data[0]; in GenericPrepare()
118 TF_LITE_ENSURE_EQ(context, op_data->input_offset, 0); in GenericPrepare()
119 TF_LITE_ENSURE_EQ(context, op_data->output_offset, 0); in GenericPrepare()
123 op_data->needs_rescale = input_scale != output_scale; in GenericPrepare()
124 if (op_name == kAbsName && op_data->needs_rescale) { in GenericPrepare()
125 SetAbsOutputMultiplier(input_scale, output_scale, &op_data->multiplier, in GenericPrepare()
126 &op_data->shift); in GenericPrepare()
128 SetRsqrtOutputMultiplier(input_scale, output_scale, &op_data->multiplier, in GenericPrepare()
[all …]
Dif.cc39 auto* op_data = new OpData; in Init() local
41 op_data->then_subgraph_index = params->then_subgraph_index; in Init()
42 op_data->else_subgraph_index = params->else_subgraph_index; in Init()
43 return op_data; in Init()
51 const OpData* op_data = reinterpret_cast<OpData*>(node->user_data); in Prepare() local
72 TF_LITE_ENSURE(context, op_data->then_subgraph_index < subgraphs->size()); in Prepare()
73 TF_LITE_ENSURE(context, op_data->else_subgraph_index < subgraphs->size()); in Prepare()
75 Subgraph* then_subgraph = (*subgraphs)[op_data->then_subgraph_index].get(); in Prepare()
76 Subgraph* else_subgraph = (*subgraphs)[op_data->else_subgraph_index].get(); in Prepare()
141 const OpData* op_data = reinterpret_cast<OpData*>(node->user_data); in Eval() local
[all …]
Dsvdf.cc64 auto* op_data = new OpData(); in Init() local
65 op_data->float_weights_time_initialized = false; in Init()
68 &op_data->scratch_tensor_index); in Init()
69 return op_data; in Init()
78 OpData* op_data = reinterpret_cast<OpData*>(node->user_data); in Prepare() local
79 int scratch_tensor_index = op_data->scratch_tensor_index; in Prepare()
169 op_data->compute_row_sums = true; in Prepare()
284 QuantizeMultiplier(effective_scale_1, &op_data->effective_scale_1_a, in Prepare()
285 &op_data->effective_scale_1_b); in Prepare()
286 QuantizeMultiplier(effective_scale_2, &op_data->effective_scale_2_a, in Prepare()
[all …]
Dlstm.cc176 OpData* op_data = static_cast<OpData*>(node->user_data); in PopulateQuantizedLstmParams8x8_16() local
177 const bool use_layer_norm = op_data->use_layer_norm; in PopulateQuantizedLstmParams8x8_16()
840 auto* op_data = new OpData(); in Init() local
841 op_data->kernel_type = kTfLiteLSTMFullKernel; in Init()
844 &op_data->scratch_tensor_index); in Init()
847 &op_data->ledger_index); in Init()
848 return op_data; in Init()
1154 OpData* op_data, in PopulatePrecomputedZPTimesWeightsWithBias() argument
1201 &op_data->integer_lstm_param; in PopulatePrecomputedZPTimesWeightsWithBias()
1213 const bool is_layer_norm = op_data->use_layer_norm; in PopulatePrecomputedZPTimesWeightsWithBias()
[all …]
Ddequantize.cc43 auto* op_data = new OpData(); in Init() local
44 op_data->float_dequantized_weights_initialized = false; in Init()
45 return op_data; in Init()
79 OpData* op_data = reinterpret_cast<OpData*>(node->user_data); in Eval() local
82 op_data->float_dequantized_weights_initialized) { in Eval()
93 op_data->float_dequantized_weights_initialized = true; in Eval()
Ddensify.cc44 auto* op_data = new OpData(); in Init() local
45 op_data->dense_weights_initialized = false; in Init()
46 return op_data; in Init()
72 OpData* op_data = reinterpret_cast<OpData*>(node->user_data); in Eval() local
74 if (op_data->dense_weights_initialized) { in Eval()
107 op_data->dense_weights_initialized = true; in Eval()
Dbasic_rnn.cc48 auto* op_data = new OpData(); in Init() local
50 &op_data->scratch_tensor_index); in Init()
51 return op_data; in Init()
112 auto* op_data = reinterpret_cast<OpData*>(node->user_data); in Prepare() local
113 op_data->compute_row_sums = true; in Prepare()
116 node->temporaries->data[0] = op_data->scratch_tensor_index; in Prepare()
127 node->temporaries->data[1] = op_data->scratch_tensor_index + 1; in Prepare()
141 node->temporaries->data[2] = op_data->scratch_tensor_index + 2; in Prepare()
154 node->temporaries->data[3] = op_data->scratch_tensor_index + 3; in Prepare()
169 node->temporaries->data[4] = op_data->scratch_tensor_index + 4; in Prepare()
[all …]
Dbatch_matmul.cc84 auto* op_data = new OpData(); in Init() local
86 op_data->rhs_transposed = false; in Init()
91 &op_data->scratch_tensor_index); in Init()
92 return op_data; in Init()
129 OpData* op_data = reinterpret_cast<OpData*>(node->user_data); in InitializeTemporaries() local
156 node->temporaries->data[0] = op_data->scratch_tensor_index; in InitializeTemporaries()
179 node->temporaries->data[1] = op_data->scratch_tensor_index + 1; in InitializeTemporaries()
220 op_data->compute_row_sums = true; in InitializeTemporaries()
221 node->temporaries->data[2] = op_data->scratch_tensor_index + 2; in InitializeTemporaries()
233 node->temporaries->data[3] = op_data->scratch_tensor_index + 3; in InitializeTemporaries()
[all …]
Dunidirectional_sequence_rnn.cc49 auto* op_data = new OpData(); in Init() local
51 &op_data->scratch_tensor_index); in Init()
52 return op_data; in Init()
119 auto* op_data = reinterpret_cast<OpData*>(node->user_data); in Prepare() local
120 op_data->compute_row_sums = true; in Prepare()
123 node->temporaries->data[0] = op_data->scratch_tensor_index; in Prepare()
134 node->temporaries->data[1] = op_data->scratch_tensor_index + 1; in Prepare()
148 node->temporaries->data[2] = op_data->scratch_tensor_index + 2; in Prepare()
161 node->temporaries->data[3] = op_data->scratch_tensor_index + 3; in Prepare()
176 node->temporaries->data[4] = op_data->scratch_tensor_index + 4; in Prepare()
[all …]
Dvar_handle.cc69 auto* op_data = static_cast<VarParams*>(node->user_data); in Eval() local
70 TF_LITE_ENSURE(context, op_data != nullptr); in Eval()
75 memcpy(output->data.raw, reinterpret_cast<char*>(&op_data->resource_id), in Eval()
76 sizeof(op_data->resource_id)); in Eval()
Dadd_n.cc40 auto* op_data = new OpData(); in Init() local
41 context->AddTensors(context, 1, &op_data->scratch_tensor_index); in Init()
42 return op_data; in Init()
62 OpData* op_data = reinterpret_cast<OpData*>(node->user_data); in Prepare() local
65 node->temporaries->data[0] = op_data->scratch_tensor_index; in Prepare()
Dbidirectional_sequence_lstm.cc164 auto* op_data = new OpData(); in Init() local
166 &op_data->scratch_tensor_index); in Init()
167 return op_data; in Init()
421 auto* op_data = reinterpret_cast<OpData*>(node->user_data); in Prepare() local
570 op_data->scratch_tensor_index + kFwScratchBuffer; in Prepare()
634 op_data->scratch_tensor_index + kBwScratchBuffer; in Prepare()
663 op_data->compute_fw_row_sums = true; in Prepare()
664 op_data->compute_bw_row_sums = true; in Prepare()
668 op_data->scratch_tensor_index + kInputQuantized; in Prepare()
681 op_data->scratch_tensor_index + kFwActivationStateQuantized; in Prepare()
[all …]
Dbidirectional_sequence_rnn.cc81 auto* op_data = new OpData(); in Init() local
83 &op_data->scratch_tensor_index); in Init()
84 return op_data; in Init()
189 OpData* op_data = reinterpret_cast<OpData*>(node->user_data); in Prepare() local
190 op_data->fw_compute_row_sums = true; in Prepare()
191 op_data->bw_compute_row_sums = true; in Prepare()
201 op_data->scratch_tensor_index + kInputQuantized; in Prepare()
214 op_data->scratch_tensor_index + kFwHiddenStateQuantized; in Prepare()
231 op_data->scratch_tensor_index + kBwHiddenStateQuantized; in Prepare()
249 op_data->scratch_tensor_index + kScalingFactors; in Prepare()
[all …]
Dbucketize.cc42 auto* op_data = new OpData(); in Init() local
45 op_data->boundaries = params->boundaries; in Init()
46 op_data->num_boundaries = params->num_boundaries; in Init()
47 return op_data; in Init()
Dunidirectional_sequence_lstm.cc89 OpData* op_data = static_cast<OpData*>(node->user_data); in PopulateQuantizedLstmParams8x8_16() local
90 const bool use_layer_norm = op_data->use_layer_norm; in PopulateQuantizedLstmParams8x8_16()
406 auto* op_data = new OpData(); in Init() local
408 &op_data->scratch_tensor_index); in Init()
409 return op_data; in Init()
714 OpData* op_data, in PopulatePrecomputedZPTimesWeightsWithBias() argument
767 &op_data->integer_lstm_param; in PopulatePrecomputedZPTimesWeightsWithBias()
781 const bool is_layer_norm = op_data->use_layer_norm; in PopulatePrecomputedZPTimesWeightsWithBias()
859 OpData* op_data = reinterpret_cast<OpData*>(node->user_data); in Prepare() local
860 const int scratch_tensor_index = op_data->scratch_tensor_index; in Prepare()
[all …]
Dsquared_difference.cc179 const auto* op_data = static_cast<const OpData*>(node->user_data); in EvalQuantizedSquaredDifference() local
182 op_data->arithmetic_params, GetTensorShape(input1), in EvalQuantizedSquaredDifference()
190 flat_size, op_data->arithmetic_params, GetTensorData<int8_t>(input1), in EvalQuantizedSquaredDifference()
/external/tensorflow/tensorflow/core/ops/
Dshape_function_test.cc26 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/
Didentify_util.cc38 const auto& op_data = op_array.GetBuffer<ArrayDataType::kFloat>().data; in CheckArrayIsScalarFloat() local
39 return op_data[0] == val; in CheckArrayIsScalarFloat()

12