Home
last modified time | relevance | path

Searched refs:kDepthwiseConv (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Densure_bias_vectors.cc37 if (op.type == OperatorType::kDepthwiseConv) { in GetOutputDepthFromWeights()
47 op.type == OperatorType::kDepthwiseConv) { in CheckOpInputSize()
82 op->type == OperatorType::kDepthwiseConv || in Run()
Dfuse_binary_into_preceding_affine.cc33 op.type == OperatorType::kDepthwiseConv) { in GetBiasIndex()
156 } else if (preceding_op->type == OperatorType::kDepthwiseConv) { in FuseMulOrDivParamsIntoPrecedingAffine()
188 } else if (preceding_op->type == OperatorType::kDepthwiseConv) { in FuseMulOrDivParamsIntoPrecedingAffine()
271 preceding_op->type != OperatorType::kDepthwiseConv && in Run()
Dfuse_binary_into_following_affine.cc94 } else if (following_op->type == OperatorType::kDepthwiseConv) { in FuseAddOrSubParamsIntoFollowingAffine()
237 following_op->type != OperatorType::kDepthwiseConv) { in Run()
274 if (following_op->type == OperatorType::kDepthwiseConv) { in Run()
Didentify_dilated_conv.cc222 conv_base_op->type == OperatorType::kDepthwiseConv) { in Run()
Dquantize.cc55 OperatorType::kDepthwiseConv, in SupportsQuantization()
239 op.type == OperatorType::kDepthwiseConv || in ChooseQuantizationForOperatorInput()
Dpropagate_fixed_sizes.cc2237 case OperatorType::kDepthwiseConv: in Run()
/external/tensorflow/tensorflow/lite/toco/tflite/
Dop_version.cc49 {{OperatorType::kDepthwiseConv, 1}, "1.5.0"}, in GetMinimumRuntimeVersionForModel()
50 {{OperatorType::kDepthwiseConv, 2}, "1.12.0"}, in GetMinimumRuntimeVersionForModel()
51 {{OperatorType::kDepthwiseConv, 3}, "1.14.0"}, in GetMinimumRuntimeVersionForModel()
52 {{OperatorType::kDepthwiseConv, 4}, "1.14.0"}, in GetMinimumRuntimeVersionForModel()
53 {{OperatorType::kDepthwiseConv, 5}, kPendingReleaseOpVersion}, in GetMinimumRuntimeVersionForModel()
Doperator_test.cc389 GetOperator("DEPTHWISE_CONV_2D", OperatorType::kDepthwiseConv), op); in TEST_F()
Doperator.cc1859 OperatorType::kDepthwiseConv)); in BuildOperatorList()
/external/tensorflow/tensorflow/lite/delegates/coreml/builders/
Dconvolution_op_builder.cc68 case ConvolutionType::kDepthwiseConv: { in Build()
130 if (conv_type_ == ConvolutionType::kDepthwiseConv) { in FillCoreMLWeights()
154 if (conv_type_ == ConvolutionType::kDepthwiseConv) { in TransposeKernelWeights()
216 case ConvolutionType::kDepthwiseConv: { in PopulateSubgraph()
291 ConvolutionType::kDepthwiseConv); in CreateDepthwiseConvolutionOpBuilder()
Dconvolution_op_builder.h27 enum class ConvolutionType { kConv, kDepthwiseConv, kTransposeConv }; enumerator
/external/tensorflow/tensorflow/lite/toco/
Ddump_graphviz.cc386 case OperatorType::kDepthwiseConv: { in GetOpAttributes()
421 if ((op.type == OperatorType::kDepthwiseConv) || in GetOpColor()
Dmodel.h50 kDepthwiseConv, enumerator
518 DepthwiseConvOperator() : Operator(OperatorType::kDepthwiseConv) {}
Dtooling_util.cc479 case OperatorType::kDepthwiseConv: in OperatorSupportsFusedActivation()
1902 case OperatorType::kDepthwiseConv: { in EstimateArithmeticOpsCount()
Dexport_tensorflow.cc2137 } else if (src_op.type == OperatorType::kDepthwiseConv) { in ConvertOperator()