Home
last modified time | relevance | path

Searched refs:depth_multiplier (Results 1 – 25 of 110) sorted by relevance

12345

/external/tensorflow/tensorflow/python/keras/applications/
Dmobilenet.py87 depth_multiplier=1, argument
210 if depth_multiplier != 1:
235 x = _depthwise_conv_block(x, 64, alpha, depth_multiplier, block_id=1)
238 x, 128, alpha, depth_multiplier, strides=(2, 2), block_id=2)
239 x = _depthwise_conv_block(x, 128, alpha, depth_multiplier, block_id=3)
242 x, 256, alpha, depth_multiplier, strides=(2, 2), block_id=4)
243 x = _depthwise_conv_block(x, 256, alpha, depth_multiplier, block_id=5)
246 x, 512, alpha, depth_multiplier, strides=(2, 2), block_id=6)
247 x = _depthwise_conv_block(x, 512, alpha, depth_multiplier, block_id=7)
248 x = _depthwise_conv_block(x, 512, alpha, depth_multiplier, block_id=8)
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
Ddepthwise_conv.h37 const int depth_multiplier = params.depth_multiplier; in DepthwiseConvPerChannel() local
58 TFLITE_DCHECK_EQ(output_depth, input_depth * depth_multiplier); in DepthwiseConvPerChannel()
65 for (int m = 0; m < depth_multiplier; ++m) { in DepthwiseConvPerChannel()
66 const int output_channel = m + in_channel * depth_multiplier; in DepthwiseConvPerChannel()
136 const int depth_multiplier = params.depth_multiplier; in DepthwiseConvPerChannel() local
155 TFLITE_DCHECK_EQ(output_depth, input_depth * depth_multiplier); in DepthwiseConvPerChannel()
162 for (int m = 0; m < depth_multiplier; ++m) { in DepthwiseConvPerChannel()
163 const int output_channel = m + in_channel * depth_multiplier; in DepthwiseConvPerChannel()
221 const int depth_multiplier = params.depth_multiplier; in DepthwiseConvHybridPerChannel() local
239 TFLITE_DCHECK_EQ(output_depth, input_depth * depth_multiplier); in DepthwiseConvHybridPerChannel()
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/
Ddepthwiseconv_per_channel_quantized_test.cc52 const int depth_multiplier = params.depth_multiplier; in PickOutputMultiplier() local
71 for (int m = 0; m < depth_multiplier; ++m) { in PickOutputMultiplier()
72 const int output_channel = m + in_channel * depth_multiplier; in PickOutputMultiplier()
144 void CompareRoundingResults(int flat_size, const int depth_multiplier, in CompareRoundingResults() argument
167 std::max(1e-2f, 1.01f / flat_size * std::sqrt(1.f * depth_multiplier)); in CompareRoundingResults()
190 int filter_width, int filter_height, int depth_multiplier, in GenerateValidShapeConfigurations() argument
203 const int output_depth = input_depth * depth_multiplier; in GenerateValidShapeConfigurations()
222 depth_multiplier, *output_shape_inference, 0); in GenerateValidShapeConfigurations()
228 const int depth_multiplier = 1; in TryTestOneDepthwiseConv3x3Filter() local
250 filter_width, filter_height, depth_multiplier, in TryTestOneDepthwiseConv3x3Filter()
[all …]
Ddepthwiseconv_per_channel_quantized_16x8_test.cc51 const int depth_multiplier = params.depth_multiplier; in PickOutputMultiplier() local
69 for (int m = 0; m < depth_multiplier; ++m) { in PickOutputMultiplier()
70 const int output_channel = m + in_channel * depth_multiplier; in PickOutputMultiplier()
142 int filter_width, int filter_height, int depth_multiplier, in GenerateValidShapeConfigurations() argument
155 const int output_depth = input_depth * depth_multiplier; in GenerateValidShapeConfigurations()
186 const int depth_multiplier = 1; in TryTestOneDepthwiseConv3x3Filter() local
205 filter_width, filter_height, depth_multiplier, in TryTestOneDepthwiseConv3x3Filter()
234 params.depth_multiplier = depth_multiplier; in TryTestOneDepthwiseConv3x3Filter()
Ddepthwiseconv_quantized_test.cc177 const int depth_multiplier = params.depth_multiplier; in DispatchDepthwiseConvImpl() local
186 pad_height, depth_multiplier, output_shape, output_shift); in DispatchDepthwiseConvImpl()
255 << " depth_multiplier = " << params.depth_multiplier in DispatchDepthwiseConvImpl()
336 << " depth_multiplier = " << params.depth_multiplier in DispatchDepthwiseConvImpl()
383 const int depth_multiplier = params.depth_multiplier; in DispatchDepthwiseConvImpl() local
392 pad_height, depth_multiplier, output_shape, output_shift); in DispatchDepthwiseConvImpl()
499 << " depth_multiplier = " << params.depth_multiplier in DispatchDepthwiseConvImpl()
620 int depth_multiplier, std::int32_t output_offset, in TestOneDepthwiseConvWithGivenOutputShift() argument
639 op_params.depth_multiplier = depth_multiplier; in TestOneDepthwiseConvWithGivenOutputShift()
697 1e-5f, 1.01f / output_buffer_size * std::sqrt(1.f * depth_multiplier)); in TestOneDepthwiseConvWithGivenOutputShift()
[all …]
Ddepthwiseconv_float_test.cc80 const int depth_multiplier = ExponentialRandomPositiveInt(0.8f, 6, 50); in TryTestOneDepthwiseConv() local
82 const int output_depth = input_depth * depth_multiplier; in TryTestOneDepthwiseConv()
138 op_params.depth_multiplier = depth_multiplier; in TryTestOneDepthwiseConv()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Ddepthwiseconv_uint8.h104 const int depth_multiplier = params.depth_multiplier; in Run() local
126 TFLITE_DCHECK_EQ(output_depth, input_depth * depth_multiplier); in Run()
133 for (int m = 0; m < depth_multiplier; m++) { in Run()
134 const int oc = m + ic * depth_multiplier; in Run()
190 const int depth_multiplier = params.depth_multiplier; in RunPerChannel() local
213 TFLITE_DCHECK_EQ(output_depth, input_depth * depth_multiplier); in RunPerChannel()
220 for (int m = 0; m < depth_multiplier; ++m) { in RunPerChannel()
221 const int output_channel = m + in_channel * depth_multiplier; in RunPerChannel()
Ddepthwiseconv_float.h37 const int depth_multiplier = params.depth_multiplier; in DepthwiseConv() local
53 TFLITE_DCHECK_EQ(output_depth, input_depth * depth_multiplier); in DepthwiseConv()
60 for (int m = 0; m < depth_multiplier; m++) { in DepthwiseConv()
61 const int oc = m + ic * depth_multiplier; in DepthwiseConv()
/external/tensorflow/tensorflow/lite/micro/kernels/vexriscv/
Ddepthwise_conv.cc50 const int depth_multiplier = params.depth_multiplier; in DepthwiseConvPerChannel() local
71 TFLITE_DCHECK_EQ(output_depth, input_depth * depth_multiplier); in DepthwiseConvPerChannel()
77 for (int m = 0; m < depth_multiplier; ++m) { in DepthwiseConvPerChannel()
106 const int output_channel = m + in_channel * depth_multiplier; in DepthwiseConvPerChannel()
136 const int output_channel = m + in_channel * depth_multiplier; in DepthwiseConvPerChannel()
172 const int depth_multiplier = params.depth_multiplier; in DepthwiseConv() local
194 TFLITE_DCHECK_EQ(output_depth, input_depth * depth_multiplier); in DepthwiseConv()
200 for (int m = 0; m < depth_multiplier; m++) { in DepthwiseConv()
229 const int output_channel = m + in_channel * depth_multiplier; in DepthwiseConv()
244 const int output_channel = m + in_channel * depth_multiplier; in DepthwiseConv()
[all …]
/external/tensorflow/tensorflow/core/kernels/neon/
Ddepthwiseconv_float.h38 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
99 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
182 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
251 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
318 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
428 int depth_multiplier, int filter_width,
445 DCHECK_EQ(depth_multiplier, kFixedDepthMultiplier);
447 DCHECK_EQ(output_depth, input_depth * depth_multiplier);
487 depth_multiplier,
499 int pad_width, int depth_multiplier, int filter_width,
[all …]
Dneon_depthwise_conv_op.cc84 const int32 depth_multiplier = filter.dim_size(3); in Compute() local
86 const int32 out_depth = in_depth * depth_multiplier; in Compute()
112 << filter_cols << ", " << in_depth << ", " << depth_multiplier in Compute()
137 bias_neon_dims, stride, pad_cols, pad_rows, depth_multiplier, in Compute()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Ddepthwiseconv_float.h34 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
95 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
178 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
249 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
319 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
425 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
458 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
493 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
526 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
578 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
[all …]
Ddepthwiseconv_uint8.h38 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
85 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
153 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
223 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
300 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
366 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
480 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
540 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
635 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
705 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
[all …]
Ddepthwiseconv_3x3_filter_common.h260 const int32 depth_multiplier = params.depth_multiplier;
268 (((input_depth % 8) == 0 && depth_multiplier == 1) ||
269 (input_depth == 1 && depth_multiplier > 1));
287 if (params.depth_multiplier == 1) {
443 int32 depth_multiplier, const RuntimeShape& output_shape,
454 filter_width == 3 && filter_height == 3 && depth_multiplier == 1 &&
/external/XNNPACK/src/subgraph/
Ddepthwise-convolution-2d.c28 uint32_t depth_multiplier, in xnn_define_depthwise_convolution_2d() argument
65 if (depth_multiplier == 0) { in xnn_define_depthwise_convolution_2d()
68 xnn_node_type_to_string(xnn_node_type_depthwise_convolution_2d), depth_multiplier); in xnn_define_depthwise_convolution_2d()
174 node->params.depthwise_convolution_2d.depth_multiplier = depth_multiplier; in xnn_define_depthwise_convolution_2d()
/external/tensorflow/tensorflow/core/kernels/
Ddepthwise_conv_op_gpu.h56 return args.depth_multiplier == 1 && args.stride == 1 && args.in_rows <= 32 &&
69 return args.depth_multiplier == 1 && args.stride == 1 && args.in_rows <= 32 &&
98 const int depth_multiplier =
99 kKnownDepthMultiplier < 0 ? args.depth_multiplier : kKnownDepthMultiplier;
114 const int in_channel = out_channel / depth_multiplier;
115 const int multiplier = out_channel % depth_multiplier;
142 depth_multiplier *
165 depth_multiplier *
345 const int depth_multiplier =
346 kKnownDepthMultiplier < 0 ? args.depth_multiplier : kKnownDepthMultiplier;
[all …]
Ddepthwise_conv_grad_op.cc107 const int64 depth_multiplier = filter_shape.dim_size(3); \
116 context, (depth_multiplier * in_depth) == out_depth, \
151 args.depth_multiplier = depth_multiplier; \
161 << in_depth << ", " << depth_multiplier << "]; stride = " << stride \
297 const int64 depth_multiplier = args.depth_multiplier; in ComputeBackpropInput() local
311 (depth_multiplier / kPacketSize) * kPacketSize; in ComputeBackpropInput()
312 const int64 dm_scalar_size = depth_multiplier % kPacketSize; in ComputeBackpropInput()
327 if (depth_multiplier == 1) { in ComputeBackpropInput()
348 if (depth_multiplier == 1) { in ComputeBackpropInput()
362 if (depth_multiplier > 1) { in ComputeBackpropInput()
[all …]
/external/tensorflow/tensorflow/python/keras/legacy_tf_layers/
Dconvolutional.py723 depth_multiplier=1, argument
746 depth_multiplier=depth_multiplier,
838 depth_multiplier=1, argument
861 depth_multiplier=depth_multiplier,
887 depth_multiplier=1, argument
982 depth_multiplier=depth_multiplier,
1010 depth_multiplier=1, argument
1110 depth_multiplier=depth_multiplier,
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
Ddepthwise_conv.h48 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
94 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
161 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
229 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
303 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
366 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
476 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
534 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
626 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
693 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
[all …]
Ddepthwise_conv_hybrid.h53 const int depth_multiplier = params.depth_multiplier; in DepthwiseConvHybridGeneral() local
88 depth_multiplier == FIXED_DEPTH_MULTIPLIER) { \ in DepthwiseConvHybridGeneral()
199 -input_offset, pad_width, depth_multiplier, filter_width, in DepthwiseConvHybridGeneral()
268 const int depth_multiplier = params.depth_multiplier; in DepthwiseConvHybridWithRounding() local
278 TFLITE_DCHECK_EQ(output_depth, input_depth * depth_multiplier); in DepthwiseConvHybridWithRounding()
296 depth_multiplier, output_shape, 0, nullptr)) { in DepthwiseConvHybridWithRounding()
/external/llvm-project/mlir/include/mlir/Dialect/Linalg/EDSC/
DBuilders.h216 int depth_multiplier = 1,
222 int depth_multiplier,
227 depth_multiplier, strides, dilations);
/external/tensorflow/tensorflow/lite/micro/kernels/
Ddepthwise_conv.cc185 op_params.depth_multiplier = params->depth_multiplier; in EvalFloat()
214 op_params.depth_multiplier = params->depth_multiplier; in EvalQuantizedPerChannel()
252 op_params.depth_multiplier = params->depth_multiplier; in EvalQuantized()
/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Ddepthwise_conv_2d_tester.h51 inline DepthwiseConv2DTester& DepthMultiplier(int32_t depth_multiplier) { in DepthMultiplier() argument
52 EXPECT_GT(depth_multiplier, 0); in DepthMultiplier()
53 depth_multiplier_ = depth_multiplier; in DepthMultiplier()
/external/tensorflow/tensorflow/lite/micro/kernels/cmsis_nn/
Ddepthwise_conv.cc228 op_params.depth_multiplier = params->depth_multiplier; in EvalFloat()
264 dw_conv_params.ch_mult = params->depth_multiplier; in EvalQuantizedPerChannel()
335 op_params.depth_multiplier = params->depth_multiplier; in EvalQuantizedPerChannel()
374 op_params.depth_multiplier = params->depth_multiplier; in EvalQuantized()
399 filter_width, filter_height, op_params.depth_multiplier, in EvalQuantized()
/external/tensorflow/tensorflow/lite/kernels/
Ddepthwise_conv.cc285 int16* depth_multiplier) { in ComputeDepthMultiplier() argument
290 *depth_multiplier = num_filter_channels / num_input_channels; in ComputeDepthMultiplier()
314 &op_params.depth_multiplier)); in EvalFloat()
358 &op_params.depth_multiplier)); in EvalQuantized()
397 &op_params.depth_multiplier)); in EvalQuantizedPerChannel()
432 op_params.depth_multiplier = params->depth_multiplier; in EvalQuantizedPerChannel16x8()
492 op_params.depth_multiplier = params->depth_multiplier; in EvalHybridPerChannel()

12345