Home
last modified time | relevance | path

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

1234

/external/tensorflow/tensorflow/lite/kernels/internal/
Ddepthwiseconv_quantized_test.cc121 const int depth_multiplier = params.depth_multiplier; in DispatchDepthwiseConv() local
130 pad_height, depth_multiplier, output_shape, output_shift); in DispatchDepthwiseConv()
184 << " depth_multiplier = " << params.depth_multiplier in DispatchDepthwiseConv()
257 << " depth_multiplier = " << params.depth_multiplier in DispatchDepthwiseConv()
294 int pad_width, int pad_height, int depth_multiplier, in TestOneDepthwiseConvWithGivenOutputShift() argument
310 op_params.depth_multiplier = depth_multiplier; in TestOneDepthwiseConvWithGivenOutputShift()
372 std::sqrt(1.f * depth_multiplier)); in TestOneDepthwiseConvWithGivenOutputShift()
427 int pad_width, int pad_height, int depth_multiplier, in TestOneDepthwiseConvBisectOutputShift() argument
439 pad_width, pad_height, depth_multiplier, output_offset, output_multiplier, in TestOneDepthwiseConvBisectOutputShift()
466 pad_width, pad_height, depth_multiplier, output_offset, output_multiplier, in TestOneDepthwiseConvBisectOutputShift()
[all …]
Ddepthwiseconv_float_test.cc77 const int depth_multiplier = ExponentialRandomPositiveInt(0.8f, 6, 50); in TryTestOneDepthwiseConv() local
79 const int output_depth = input_depth * depth_multiplier; in TryTestOneDepthwiseConv()
135 op_params.depth_multiplier = depth_multiplier; in TryTestOneDepthwiseConv()
/external/tensorflow/tensorflow/core/kernels/
Ddepthwise_conv_op.h33 int depth_multiplier; member
50 depth_multiplier(0), in DepthwiseArgs()
207 (args.depth_multiplier / kPacketSize) * kPacketSize;
208 const int64 dm_scalar_size = args.depth_multiplier % kPacketSize;
215 const int64 replicated_packet_size = kPacketSize * args.depth_multiplier;
236 for (int dm = 0; dm < args.depth_multiplier; ++dm) {
238 args.depth_multiplier);
246 const int64 base = d * args.depth_multiplier;
261 for (int dm = 0; dm < args.depth_multiplier; ++dm) {
266 in_buf += input_scalar_size * args.depth_multiplier;
Ddepthwise_conv_op_gpu.h46 return args.depth_multiplier == 1 && args.stride == 1 && args.in_rows <= 32 && in CanLaunchDepthwiseConv2dGPUSmall()
59 return args.depth_multiplier == 1 && args.stride == 1 && args.in_rows <= 32 && in CanLaunchDepthwiseConv2dBackpropFilterGPUSmall()
85 const int depth_multiplier = in DepthwiseConv2dGPUKernelNHWC() local
86 kKnownDepthMultiplier < 0 ? args.depth_multiplier : kKnownDepthMultiplier; in DepthwiseConv2dGPUKernelNHWC()
101 const int in_channel = out_channel / depth_multiplier; in DepthwiseConv2dGPUKernelNHWC()
102 const int multiplier = out_channel % depth_multiplier; in DepthwiseConv2dGPUKernelNHWC()
129 depth_multiplier * in DepthwiseConv2dGPUKernelNHWC()
151 depth_multiplier * in DepthwiseConv2dGPUKernelNHWC()
325 const int depth_multiplier = in DepthwiseConv2dGPUKernelNCHW() local
326 kKnownDepthMultiplier < 0 ? args.depth_multiplier : kKnownDepthMultiplier; in DepthwiseConv2dGPUKernelNCHW()
[all …]
Ddepthwise_conv_grad_op.cc102 const int64 depth_multiplier = filter_shape.dim_size(3); \
111 context, (depth_multiplier * in_depth) == out_depth, \
139 args.depth_multiplier = depth_multiplier; \
149 << in_depth << ", " << depth_multiplier << "]; stride = " << stride \
285 const int64 depth_multiplier = args.depth_multiplier; in ComputeBackpropInput() local
299 (depth_multiplier / kPacketSize) * kPacketSize; in ComputeBackpropInput()
300 const int64 dm_scalar_size = depth_multiplier % kPacketSize; in ComputeBackpropInput()
315 if (depth_multiplier == 1) { in ComputeBackpropInput()
336 if (depth_multiplier == 1) { in ComputeBackpropInput()
350 if (depth_multiplier > 1) { in ComputeBackpropInput()
[all …]
Ddepthwise_conv_op.cc324 const int32 depth_multiplier = filter.dim_size(3); in Compute() local
327 const int32 out_depth = in_depth * depth_multiplier; in Compute()
380 << filter_cols << ", " << in_depth << ", " << depth_multiplier in Compute()
419 args.depth_multiplier = 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.h37 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
84 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
152 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
222 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
299 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
365 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
479 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
539 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
634 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
704 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
[all …]
Dlegacy_optimized_ops.h143 int pad_width, int pad_height, int depth_multiplier, in DepthwiseConv() argument
156 op_params.depth_multiplier = depth_multiplier; in DepthwiseConv()
169 int pad_height, int depth_multiplier, in DepthwiseConv() argument
175 pad_height, depth_multiplier, output_activation_min, in DepthwiseConv()
185 int pad_height, int depth_multiplier, float* output_data, in DepthwiseConv() argument
191 depth_multiplier, output_activation_min, output_activation_max, in DepthwiseConv()
200 int pad_width, int pad_height, int depth_multiplier, in DepthwiseConv() argument
204 depth_multiplier, output_data, output_dims); in DepthwiseConv()
213 int pad_width, int pad_height, int depth_multiplier, in DepthwiseConv() argument
227 op_params.depth_multiplier = 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);
489 depth_multiplier,
501 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/reference/
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()
Ddepthwiseconv_uint8.h105 const int depth_multiplier = params.depth_multiplier; in Run() local
127 TFLITE_DCHECK_EQ(output_depth, input_depth * depth_multiplier); in Run()
134 for (int m = 0; m < depth_multiplier; m++) { in Run()
135 const int oc = m + ic * depth_multiplier; in Run()
Dlegacy_reference_ops.h44 int pad_width, int pad_height, int depth_multiplier, in DepthwiseConv() argument
57 op_params.depth_multiplier = depth_multiplier; in DepthwiseConv()
70 int pad_height, int depth_multiplier, in DepthwiseConv() argument
76 pad_height, depth_multiplier, output_activation_min, in DepthwiseConv()
86 int pad_height, int depth_multiplier, float* output_data, in DepthwiseConv() argument
92 depth_multiplier, output_activation_min, output_activation_max, in DepthwiseConv()
101 int pad_width, int pad_height, int depth_multiplier, in DepthwiseConv() argument
105 depth_multiplier, output_data, output_dims); in DepthwiseConv()
114 int pad_width, int pad_height, int depth_multiplier, in DepthwiseConv() argument
128 op_params.depth_multiplier = depth_multiplier; in DepthwiseConv()
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
Ddepthwise_conv.h36 const int depth_multiplier = params.depth_multiplier; in DepthwiseConvPerChannel() local
59 TFLITE_DCHECK_EQ(output_depth, input_depth * depth_multiplier); in DepthwiseConvPerChannel()
66 for (int m = 0; m < depth_multiplier; ++m) { in DepthwiseConvPerChannel()
67 const int output_channel = m + in_channel * depth_multiplier; in DepthwiseConvPerChannel()
/external/tensorflow/tensorflow/python/layers/
Dconvolutional.py701 depth_multiplier=1, argument
724 depth_multiplier=depth_multiplier,
813 depth_multiplier=1, argument
836 depth_multiplier=depth_multiplier,
865 depth_multiplier=1, argument
954 depth_multiplier=depth_multiplier,
985 depth_multiplier=1, argument
1079 depth_multiplier=depth_multiplier,
/external/tensorflow/tensorflow/lite/experimental/micro/kernels/portable_optimized/
Ddepthwise_conv.cc105 const int depth_multiplier = params.depth_multiplier; in DepthwiseConvOptimizedForFilterWidthEight() local
127 TFLITE_DCHECK_EQ(output_depth, input_depth * depth_multiplier); in DepthwiseConvOptimizedForFilterWidthEight()
168 for (int m = 0; m < depth_multiplier; m++) { in DepthwiseConvOptimizedForFilterWidthEight()
169 const int oc = m + ic * depth_multiplier; in DepthwiseConvOptimizedForFilterWidthEight()
302 op_params.depth_multiplier = params->depth_multiplier; in EvalFloat()
330 op_params.depth_multiplier = params->depth_multiplier; in EvalQuantized()
/external/tensorflow/tensorflow/lite/kernels/
Ddepthwise_conv.cc103 params->depth_multiplier * SizeOfDimension(input, 3), in Prepare()
215 op_params.depth_multiplier = params->depth_multiplier; in EvalFloat()
252 op_params.depth_multiplier = params->depth_multiplier; in EvalQuantized()
280 op_params.depth_multiplier = params->depth_multiplier; in EvalQuantizedPerChannel()
/external/tensorflow/tensorflow/contrib/slim/python/slim/nets/
Dinception_v2.py38 depth_multiplier=1.0, argument
77 if depth_multiplier <= 0:
79 depth = lambda d: max(int(d * depth_multiplier), min_depth)
106 depth_multiplier=depthwise_multiplier,
488 depth_multiplier=1.0, argument
534 if depth_multiplier <= 0:
546 depth_multiplier=depth_multiplier)
Dinception_v3.py38 depth_multiplier=1.0, argument
101 if depth_multiplier <= 0:
103 depth = lambda d: max(int(d * depth_multiplier), min_depth)
518 depth_multiplier=1.0, argument
568 if depth_multiplier <= 0:
570 depth = lambda d: max(int(d * depth_multiplier), min_depth)
580 depth_multiplier=depth_multiplier)
Dinception_v2_test.py140 inputs, num_classes, scope='depth_multiplied_net', depth_multiplier=0.5)
161 inputs, num_classes, scope='depth_multiplied_net', depth_multiplier=2.0)
175 _ = inception_v2.inception_v2(inputs, num_classes, depth_multiplier=-0.1)
177 _ = inception_v2.inception_v2(inputs, num_classes, depth_multiplier=0.0)
Dinception_v3_test.py170 inputs, num_classes, scope='depth_multiplied_net', depth_multiplier=0.5)
191 inputs, num_classes, scope='depth_multiplied_net', depth_multiplier=2.0)
205 _ = inception_v3.inception_v3(inputs, num_classes, depth_multiplier=-0.1)
207 _ = inception_v3.inception_v3(inputs, num_classes, depth_multiplier=0.0)
/external/tensorflow/tensorflow/lite/experimental/micro/kernels/
Ddepthwise_conv.cc111 op_params.depth_multiplier = params->depth_multiplier; in EvalFloat()
139 op_params.depth_multiplier = params->depth_multiplier; in EvalQuantized()
/external/tensorflow/tensorflow/python/keras/layers/
Dconvolutional.py1236 depth_multiplier=1, argument
1269 self.depth_multiplier = depth_multiplier
1290 self.depth_multiplier)
1292 1,) * self.rank + (self.depth_multiplier * input_dim, self.filters)
1339 self.depth_multiplier,
1439 depth_multiplier=1, argument
1461 depth_multiplier=depth_multiplier,
1608 depth_multiplier=1, argument
1630 depth_multiplier=depth_multiplier,
1740 depth_multiplier=1, argument
[all …]
/external/tensorflow/tensorflow/contrib/tensor_forest/kernels/v4/
Dparams.cc39 param.exponential().depth_multiplier() * depth)); in ResolveParam()

1234