Cross Reference: /external/tensorflow/tensorflow/contrib/lite/toco/tflite/operator

Lines Matching refs:op
59   std::unique_ptr<T> SerializeAndDeserialize(const BaseOperator& op,
63     Options input_options = op.Serialize(toco_op, &builder);
74     auto new_toco_op = op.Deserialize(output_options->builtin_options(),
113   AddOperator op;
114   op.fused_activation_function = FusedActivationFunctionType::kRelu6;
116       SerializeAndDeserialize(GetOperator("ADD", OperatorType::kAdd), op);
117   EXPECT_EQ(op.fused_activation_function,
122   MeanOperator op;
123   op.keep_dims = false;
126       SerializeAndDeserialize(GetOperator("MEAN", OperatorType::kMean), op);
127   EXPECT_EQ(op.keep_dims, output_toco_op->keep_dims);
131   CastOperator op;
132   op.src_data_type = ArrayDataType::kFloat;
133   op.dst_data_type = ArrayDataType::kUint8;
135       SerializeAndDeserialize(GetOperator("CAST", OperatorType::kCast), op);
136   EXPECT_EQ(op.src_data_type, output_toco_op->src_data_type);
137   EXPECT_EQ(op.dst_data_type, output_toco_op->dst_data_type);
141   ConcatenationOperator op;
142   op.axis = 123;
144       GetOperator("CONCATENATION", OperatorType::kConcatenation), op);
145   EXPECT_EQ(op.axis, output_toco_op->axis);
149   DepthToSpaceOperator op;
150   op.block_size = 123;
152       GetOperator("DEPTH_TO_SPACE", OperatorType::kDepthToSpace), op);
153   EXPECT_EQ(op.block_size, output_toco_op->block_size);
157   FakeQuantOperator op;
161   op.minmax.reset(minmax);
163       GetOperator("FAKE_QUANT", OperatorType::kFakeQuant), op);
164   EXPECT_EQ(op.minmax->min, output_toco_op->minmax->min);
165   EXPECT_EQ(op.minmax->max, output_toco_op->minmax->max);
169   FullyConnectedOperator op;
170   op.fused_activation_function = FusedActivationFunctionType::kRelu6;
172       GetOperator("FULLY_CONNECTED", OperatorType::kFullyConnected), op);
173   EXPECT_EQ(op.fused_activation_function,
178   GatherOperator op;
180       SerializeAndDeserialize(GetOperator("GATHER", OperatorType::kGather), op);
185   L2PoolOperator op;
186   op.stride_width = 123;
187   op.stride_height = 124;
188   op.padding.type = PaddingType::kValid;
189   op.kwidth = 480;
190   op.kheight = 1080;
192       GetOperator("L2_POOL_2D", OperatorType::kL2Pool), op);
193   EXPECT_EQ(op.stride_width, output_toco_op->stride_width);
194   EXPECT_EQ(op.stride_height, output_toco_op->stride_height);
195   EXPECT_EQ(op.padding.type, output_toco_op->padding.type);
196   EXPECT_EQ(op.kwidth, output_toco_op->kwidth);
197   EXPECT_EQ(op.kheight, output_toco_op->kheight);
201   LocalResponseNormalizationOperator op;
202   op.range = 123;
203   op.bias = 1.23;
204   op.alpha = 12.3;
205   op.beta = .123;
209       op);
210   EXPECT_EQ(op.range, output_toco_op->range);
211   EXPECT_EQ(op.bias, output_toco_op->bias);
212   EXPECT_EQ(op.alpha, output_toco_op->alpha);
213   EXPECT_EQ(op.beta, output_toco_op->beta);
217   MaxPoolOperator op;
218   op.stride_width = 123;
219   op.stride_height = 124;
220   op.padding.type = PaddingType::kValid;
221   op.kwidth = 480;
222   op.kheight = 1080;
224       GetOperator("MAX_POOL_2D", OperatorType::kMaxPool), op);
225   EXPECT_EQ(op.stride_width, output_toco_op->stride_width);
226   EXPECT_EQ(op.stride_height, output_toco_op->stride_height);
227   EXPECT_EQ(op.padding.type, output_toco_op->padding.type);
228   EXPECT_EQ(op.kwidth, output_toco_op->kwidth);
229   EXPECT_EQ(op.kheight, output_toco_op->kheight);
233   TensorFlowReshapeOperator op;
234   op.shape = {1, 2, 4, 5, 8};
236       GetOperator("RESHAPE", OperatorType::kTensorFlowReshape), op);
237   EXPECT_EQ(op.shape, output_toco_op->shape);
241   SoftmaxOperator op;
242   op.beta = 123.1;
244       GetOperator("SOFTMAX", OperatorType::kSoftmax), op);
245   EXPECT_EQ(op.beta, output_toco_op->beta);
249   SpaceToDepthOperator op;
250   op.block_size = 123;
252       GetOperator("SPACE_TO_DEPTH", OperatorType::kSpaceToDepth), op);
253   EXPECT_EQ(op.block_size, output_toco_op->block_size);
257   TensorFlowSplitOperator op;
258   op.num_split = 123;
260       GetOperator("SPLIT", OperatorType::kTensorFlowSplit), op);
261   EXPECT_EQ(op.num_split, output_toco_op->num_split);
265   AveragePoolOperator op;
266   op.fused_activation_function = FusedActivationFunctionType::kRelu6;
267   op.stride_width = 123;
268   op.stride_height = 124;
269   op.padding.type = PaddingType::kValid;
270   op.kwidth = 480;
271   op.kheight = 1080;
273       GetOperator("AVERAGE_POOL_2D", OperatorType::kAveragePool), op);
274   EXPECT_EQ(op.fused_activation_function,
276   EXPECT_EQ(op.stride_width, output_toco_op->stride_width);
277   EXPECT_EQ(op.stride_height, output_toco_op->stride_height);
278   EXPECT_EQ(op.padding.type, output_toco_op->padding.type);
279   EXPECT_EQ(op.kwidth, output_toco_op->kwidth);
280   EXPECT_EQ(op.kheight, output_toco_op->kheight);
284   ConvOperator op;
285   op.stride_width = 123;
286   op.stride_height = 124;
287   op.padding.type = PaddingType::kValid;
288   op.fused_activation_function = FusedActivationFunctionType::kRelu6;
290       SerializeAndDeserialize(GetOperator("CONV_2D", OperatorType::kConv), op);
291   EXPECT_EQ(op.stride_width, output_toco_op->stride_width);
292   EXPECT_EQ(op.stride_height, output_toco_op->stride_height);
293   EXPECT_EQ(op.padding.type, output_toco_op->padding.type);
294   EXPECT_EQ(op.fused_activation_function,
299   DepthwiseConvOperator op;
300   op.stride_width = 123;
301   op.stride_height = 124;
302   op.padding.type = PaddingType::kValid;
303   op.depth_multiplier = 6;
304   op.fused_activation_function = FusedActivationFunctionType::kRelu6;
306       GetOperator("DEPTHWISE_CONV_2D", OperatorType::kDepthwiseConv), op);
307   EXPECT_EQ(op.stride_width, output_toco_op->stride_width);
308   EXPECT_EQ(op.stride_height, output_toco_op->stride_height);
309   EXPECT_EQ(op.padding.type, output_toco_op->padding.type);
310   EXPECT_EQ(op.depth_multiplier, output_toco_op->depth_multiplier);
311   EXPECT_EQ(op.fused_activation_function,
316   L2NormalizationOperator op;
317   op.fused_activation_function = FusedActivationFunctionType::kRelu6;
319       GetOperator("L2_NORMALIZATION", OperatorType::kL2Normalization), op);
320   EXPECT_EQ(op.fused_activation_function,
325   MulOperator op;
326   op.fused_activation_function = FusedActivationFunctionType::kRelu6;
328       SerializeAndDeserialize(GetOperator("MUL", OperatorType::kMul), op);
329   EXPECT_EQ(op.fused_activation_function,
334   ResizeBilinearOperator op;
335   op.align_corners = true;
337       GetOperator("RESIZE_BILINEAR", OperatorType::kResizeBilinear), op);
338   EXPECT_EQ(op.align_corners, output_toco_op->align_corners);
342   SvdfOperator op;
343   op.fused_activation_function = FusedActivationFunctionType::kRelu;
344   op.rank = 1;
346       SerializeAndDeserialize(GetOperator("SVDF", OperatorType::kSvdf), op);
347   EXPECT_EQ(op.fused_activation_function,
349   EXPECT_EQ(op.rank, output_toco_op->rank);
353   SqueezeOperator op;
354   op.squeeze_dims = {-2, -3, 4, 1, 4};
357       GetOperator("SQUEEZE", OperatorType::kSqueeze), op);
358   EXPECT_EQ(op.squeeze_dims, output_toco_op->squeeze_dims);
362   StridedSliceOperator op;
364   op.begin_mask = 1;
365   op.end_mask = 2;
366   op.ellipsis_mask = 1;
367   op.new_axis_mask = 1;
368   op.shrink_axis_mask = 2;
371       GetOperator("STRIDED_SLICE", OperatorType::kStridedSlice), op);
372   EXPECT_EQ(op.start_indices, output_toco_op->start_indices);
373   EXPECT_EQ(op.stop_indices, output_toco_op->stop_indices);
374   EXPECT_EQ(op.strides, output_toco_op->strides);
375   EXPECT_EQ(op.begin_mask, output_toco_op->begin_mask);
376   EXPECT_EQ(op.end_mask, output_toco_op->end_mask);
377   EXPECT_EQ(op.end_mask, output_toco_op->end_mask);
378   EXPECT_EQ(op.ellipsis_mask, output_toco_op->ellipsis_mask);
379   EXPECT_EQ(op.new_axis_mask, output_toco_op->new_axis_mask);
380   EXPECT_EQ(op.shrink_axis_mask, output_toco_op->shrink_axis_mask);
384   TopKV2Operator op;
386       GetOperator("TOPK_V2", OperatorType::kTopK_V2), op);
391   TensorFlowUnsupportedOperator op;
392   op.tensorflow_op = "MyCustomUnsupportedOp";
400   node_def.SerializeToString(&op.tensorflow_node_def);
405                               op);
417   TensorFlowUnsupportedOperator op;
418   op.tensorflow_op = "MyCustomUnsupportedOp";
422                               op);