Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/python/keras/_impl/keras/applications/
Dmobilenet.py187 depth_multiplier=1, argument
211 self.depth_multiplier = depth_multiplier
232 input_dim, self.depth_multiplier)
243 shape=(input_dim * self.depth_multiplier,),
276 out_filters = input_shape[1] * self.depth_multiplier
280 out_filters = input_shape[3] * self.depth_multiplier
298 config['depth_multiplier'] = self.depth_multiplier
312 depth_multiplier=1, argument
433 if depth_multiplier != 1:
472 x = _depthwise_conv_block(x, 64, alpha, depth_multiplier, block_id=1)
[all …]
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/reference/
Ddepthwiseconv_float.h29 int pad_height, int depth_multiplier, in DepthwiseConv() argument
42 TFLITE_DCHECK(output_depth == input_depth * depth_multiplier); in DepthwiseConv()
48 for (int m = 0; m < depth_multiplier; m++) { in DepthwiseConv()
49 const int oc = m + ic * depth_multiplier; in DepthwiseConv()
90 int pad_height, int depth_multiplier, float* output_data, in DepthwiseConv() argument
96 depth_multiplier, output_activation_min, output_activation_max, in DepthwiseConv()
105 int pad_width, int pad_height, int depth_multiplier, in DepthwiseConv() argument
109 depth_multiplier, output_data, output_dims); in DepthwiseConv()
Ddepthwiseconv_uint8.h34 int pad_height, int depth_multiplier, in DepthwiseConv() argument
48 TFLITE_DCHECK(output_depth == input_depth * depth_multiplier); in DepthwiseConv()
54 for (int m = 0; m < depth_multiplier; m++) { in DepthwiseConv()
55 const int oc = m + ic * depth_multiplier; in DepthwiseConv()
100 int pad_height, int depth_multiplier, int32 output_offset, in DepthwiseConv() argument
110 stride_height, pad_width, pad_height, depth_multiplier, in DepthwiseConv()
122 int pad_width, int pad_height, int depth_multiplier, in DepthwiseConv() argument
129 stride, pad_width, pad_height, depth_multiplier, in DepthwiseConv()
/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.cu.cc43 return args.depth_multiplier == 1 && args.stride == 1 && args.in_rows <= 32 && in CanLaunchDepthwiseConv2dGPUSmall()
56 return args.depth_multiplier == 1 && args.stride == 1 && args.in_rows <= 32 && in CanLaunchDepthwiseConv2dBackpropFilterGPUSmall()
82 const int depth_multiplier = in DepthwiseConv2dGPUKernelNHWC() local
83 kKnownDepthMultiplier < 0 ? args.depth_multiplier : kKnownDepthMultiplier; in DepthwiseConv2dGPUKernelNHWC()
98 const int in_channel = out_channel / depth_multiplier; in DepthwiseConv2dGPUKernelNHWC()
99 const int multiplier = out_channel % depth_multiplier; in DepthwiseConv2dGPUKernelNHWC()
126 depth_multiplier * in DepthwiseConv2dGPUKernelNHWC()
148 depth_multiplier * in DepthwiseConv2dGPUKernelNHWC()
318 const int depth_multiplier = in DepthwiseConv2dGPUKernelNCHW() local
319 kKnownDepthMultiplier < 0 ? args.depth_multiplier : kKnownDepthMultiplier; in DepthwiseConv2dGPUKernelNCHW()
[all …]
Ddepthwise_conv_grad_op.cc99 const int64 depth_multiplier = filter_shape.dim_size(3); \
108 context, (depth_multiplier * in_depth) == out_depth, \
136 args.depth_multiplier = depth_multiplier; \
146 << in_depth << ", " << depth_multiplier << "]; stride = " << stride \
282 const int64 depth_multiplier = args.depth_multiplier; in ComputeBackpropInput() local
296 (depth_multiplier / kPacketSize) * kPacketSize; in ComputeBackpropInput()
297 const int64 dm_scalar_size = depth_multiplier % kPacketSize; in ComputeBackpropInput()
312 if (depth_multiplier == 1) { in ComputeBackpropInput()
333 if (depth_multiplier == 1) { in ComputeBackpropInput()
347 if (depth_multiplier > 1) { in ComputeBackpropInput()
[all …]
Ddepthwise_conv_op.cc317 const int32 depth_multiplier = filter.dim_size(3); in Compute() local
320 const int32 out_depth = in_depth * depth_multiplier; in Compute()
363 << filter_cols << ", " << in_depth << ", " << depth_multiplier in Compute()
391 args.depth_multiplier = depth_multiplier; in Compute()
/external/tensorflow/tensorflow/contrib/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.h34 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
81 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
149 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
219 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
296 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
362 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,
536 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
631 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
701 static void Run(int num_output_pixels, int input_depth, int depth_multiplier,
[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/python/layers/
Dconvolutional.py901 depth_multiplier=1, argument
933 self.depth_multiplier = depth_multiplier
954 self.depth_multiplier)
956 1,) * self.rank + (self.depth_multiplier * input_dim, self.filters)
1056 depth_multiplier=1, argument
1080 depth_multiplier=depth_multiplier,
1205 depth_multiplier=1, argument
1229 depth_multiplier=depth_multiplier,
1280 depth_multiplier=1, argument
1369 depth_multiplier=depth_multiplier,
[all …]
/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/contrib/lite/kernels/
Ddepthwise_conv.cc99 params->depth_multiplier * SizeOfDimension(input, 3), in Prepare()
192 params->depth_multiplier, output_activation_min, output_activation_max, in EvalFloat()
220 params->depth_multiplier, output_offset, data->output_multiplier, in EvalQuantized()
/external/tensorflow/tensorflow/tools/api/golden/
Dtensorflow.keras.applications.mobilenet.pbtxt5 …argspec: "args=[\'input_shape\', \'alpha\', \'depth_multiplier\', \'dropout\', \'include_top\', \'…
Dtensorflow.layers.pbtxt145 …\', \'strides\', \'padding\', \'data_format\', \'dilation_rate\', \'depth_multiplier\', \'activati…
149 …\', \'strides\', \'padding\', \'data_format\', \'dilation_rate\', \'depth_multiplier\', \'activati…
/external/tensorflow/tensorflow/contrib/tensor_forest/kernels/v4/
Dparams.cc39 param.exponential().depth_multiplier() * depth)); in ResolveParam()
/external/tensorflow/tensorflow/contrib/lite/
Dbuiltin_op_data.h73 int depth_multiplier; member
/external/tensorflow/tensorflow/contrib/lite/toco/tflite/
Doperator_test.cc303 op.depth_multiplier = 6; in TEST_F()
310 EXPECT_EQ(op.depth_multiplier, output_toco_op->depth_multiplier); in TEST_F()
/external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/
Dconvert_pure_conv_to_depthwise.cc70 depthwiseconv_op->depth_multiplier = weights_array.shape().dims(0); in Run()
/external/tensorflow/tensorflow/contrib/quantize/python/
Dquantize_test.py100 depth_multiplier=1.0, padding='SAME',
/external/tensorflow/tensorflow/compiler/xla/
Dreference_util.cc156 const int64 depth_multiplier = depthwise_weights.planes(); in SeparableConvArray4D() local
157 CHECK_EQ(pointwise_weights.depth(), input.depth() * depth_multiplier); in SeparableConvArray4D()
168 for (int64 depth = 0; depth < depth_multiplier; ++depth) { in SeparableConvArray4D()
171 pointwise_weights(out, depth + kz * depth_multiplier, 0, 0); in SeparableConvArray4D()

123