Home
last modified time | relevance | path

Searched defs:depth_multiplier (Results 1 – 25 of 31) sorted by relevance

12

/external/tensorflow/tensorflow/python/keras/applications/
Dmobilenet.py86 depth_multiplier=1, argument
354 depth_multiplier=1, argument
/external/tensorflow/tensorflow/lite/kernels/internal/
Ddepthwiseconv_per_channel_quantized_test.cc52 const int depth_multiplier = params.depth_multiplier; in PickOutputMultiplier() local
144 void CompareRoundingResults(int flat_size, const int depth_multiplier, in CompareRoundingResults()
191 int filter_width, int filter_height, int depth_multiplier, in GenerateValidShapeConfigurations()
229 const int depth_multiplier = 1; in TryTestOneDepthwiseConv3x3Filter() local
Ddepthwiseconv_quantized_test.cc177 const int depth_multiplier = params.depth_multiplier; in DispatchDepthwiseConvImpl() local
383 const int depth_multiplier = params.depth_multiplier; in DispatchDepthwiseConvImpl() local
581 int depth_multiplier, std::int32_t output_offset, in TestOneDepthwiseConvWithGivenOutputShift()
718 int depth_multiplier, std::int32_t output_offset, in TestOneDepthwiseConvBisectOutputShift()
777 int depth_multiplier, std::int32_t output_offset, in TestOneDepthwiseConv()
794 int depth_multiplier, int stride, in TryTestDepthwiseConv()
924 const int depth_multiplier = ExponentialRandomPositiveInt(0.8f, 6, 50); in TryTestOneDepthwiseConv() local
946 const int depth_multiplier = 1; in TryTestOneDepthwiseConv3x3Filter() local
998 const int depth_multiplier = in TryTestOneNeonDot3x3() local
Ddepthwiseconv_float_test.cc80 const int depth_multiplier = ExponentialRandomPositiveInt(0.8f, 6, 50); in TryTestOneDepthwiseConv() local
/external/tensorflow/tensorflow/core/kernels/
Ddepthwise_conv_op.h33 int depth_multiplier; member
Ddepthwise_conv_op.cc335 const int32 depth_multiplier = filter.dim_size(3); in Compute() local
Ddepthwise_conv_grad_op.cc289 const int64 depth_multiplier = args.depth_multiplier; in ComputeBackpropInput() local
/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
Ddepthwise_conv.h37 const int depth_multiplier = params.depth_multiplier; in DepthwiseConvPerChannel() local
135 const int depth_multiplier = params.depth_multiplier; in DepthwiseConvHybridPerChannel() local
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Ddepthwiseconv_uint8.h105 const int depth_multiplier = params.depth_multiplier; in Run() local
191 const int depth_multiplier = params.depth_multiplier; in RunPerChannel() local
Ddepthwiseconv_float.h37 const int depth_multiplier = params.depth_multiplier; in DepthwiseConv() local
Dlegacy_reference_ops.h46 int pad_width, int pad_height, int depth_multiplier, in DepthwiseConv()
72 int pad_height, int depth_multiplier, in DepthwiseConv()
88 int pad_height, int depth_multiplier, float* output_data, in DepthwiseConv()
103 int pad_width, int pad_height, int depth_multiplier, in DepthwiseConv()
116 int pad_width, int pad_height, int depth_multiplier, in DepthwiseConv()
150 int pad_height, int depth_multiplier, in DepthwiseConv()
170 int pad_height, int depth_multiplier, int32 output_offset, in DepthwiseConv()
192 int pad_width, int pad_height, int depth_multiplier, in DepthwiseConv()
/external/tensorflow/tensorflow/core/kernels/neon/
Dneon_depthwise_conv_op.cc84 const int32 depth_multiplier = filter.dim_size(3); in Compute() local
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h387 const dnn::FilterDescriptor& filter_descriptor, int depth_multiplier, in DoSeparableConvolve()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h372 const dnn::FilterDescriptor& filter_descriptor, int depth_multiplier, in DoSeparableConvolve()
/external/tensorflow/tensorflow/python/layers/
Dconvolutional.py701 depth_multiplier=1, argument
813 depth_multiplier=1, argument
865 depth_multiplier=1, argument
985 depth_multiplier=1, argument
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
Ddepthwise_conv_hybrid.h53 const int depth_multiplier = params.depth_multiplier; in DepthwiseConvHybridGeneral() local
268 const int depth_multiplier = params.depth_multiplier; in DepthwiseConvHybridWithRounding() local
/external/tensorflow/tensorflow/python/keras/layers/
Dconvolutional.py1412 depth_multiplier=1, argument
1640 depth_multiplier=1, argument
1825 depth_multiplier=1, argument
1971 depth_multiplier=1, argument
/external/XNNPACK/src/
Dsubgraph.c278 uint32_t depth_multiplier, in xnn_define_depthwise_convolution_2d()
/external/tensorflow/tensorflow/lite/micro/kernels/xtensa-hifimini/
Ddepthwise_conv.cc52 const int depth_multiplier = params.depth_multiplier; in DepthwiseConvPerChannel() local
/external/tensorflow/tensorflow/lite/kernels/
Ddepthwise_conv.cc265 int16* depth_multiplier) { in ComputeDepthMultiplier()
/external/XNNPACK/src/xnnpack/
Dsubgraph.h102 uint32_t depth_multiplier; member
/external/tensorflow/tensorflow/lite/micro/kernels/portable_optimized/
Ddepthwise_conv.cc121 const int depth_multiplier = params.depth_multiplier; in DepthwiseConvOptimizedForFilterWidthEight() local
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h144 int pad_width, int pad_height, int depth_multiplier, in DepthwiseConv()
175 int pad_height, int depth_multiplier, in DepthwiseConv()
191 int pad_height, int depth_multiplier, float* output_data, in DepthwiseConv()
206 int pad_width, int pad_height, int depth_multiplier, in DepthwiseConv()
221 const int depth_multiplier = params.depth_multiplier; in LegacyDepthwiseConvWithRounding() local
287 int pad_width, int pad_height, int depth_multiplier, in DepthwiseConv()
326 int pad_height, int depth_multiplier, in DepthwiseConv()
346 int pad_height, int depth_multiplier, int32 output_offset, in DepthwiseConv()
368 int pad_width, int pad_height, int depth_multiplier, in DepthwiseConv()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dmodel_builder.cc943 const int depth_multiplier = tf_options->depth_multiplier; in IsSupported() local
999 const int depth_multiplier = tf_options->depth_multiplier; in Parse() local
1017 const TfLiteTensor* output, int depth_multiplier, in TransposeWeights()
/external/tensorflow/tensorflow/compiler/xla/
Dreference_util.cc108 const int64 depth_multiplier = depthwise_weights.planes(); in SeparableConvArray4D() local

12