Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/contrib/slim/python/slim/nets/
Dinception_v2.py38 depth_multiplier=1.0, argument
488 depth_multiplier=1.0, argument
Dinception_v3.py38 depth_multiplier=1.0, argument
518 depth_multiplier=1.0, argument
/external/tensorflow/tensorflow/lite/kernels/internal/
Ddepthwiseconv_quantized_test.cc121 const int depth_multiplier = params.depth_multiplier; in DispatchDepthwiseConv() local
294 int pad_width, int pad_height, int depth_multiplier, in TestOneDepthwiseConvWithGivenOutputShift()
427 int pad_width, int pad_height, int depth_multiplier, in TestOneDepthwiseConvBisectOutputShift()
477 int pad_width, int pad_height, int depth_multiplier, in TestOneDepthwiseConv()
492 int depth_multiplier, int stride, in TryTestDepthwiseConv()
567 const int depth_multiplier = ExponentialRandomPositiveInt(0.8f, 6, 50); in TryTestOneDepthwiseConv() local
589 const int depth_multiplier = 1; in TryTestOneDepthwiseConv3x3Filter() local
641 const int depth_multiplier = in TryTestOneNeonDot3x3() local
Ddepthwiseconv_float_test.cc77 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.cc324 const int32 depth_multiplier = filter.dim_size(3); in Compute() local
Ddepthwise_conv_op_gpu.h85 const int depth_multiplier = in DepthwiseConv2dGPUKernelNHWC() local
325 const int depth_multiplier = in DepthwiseConv2dGPUKernelNCHW() local
Ddepthwise_conv_grad_op.cc285 const int64 depth_multiplier = args.depth_multiplier; in ComputeBackpropInput() local
Dnn_ops_test.cc499 int in_depth, int depth_multiplier, in BM_ConvFloatDepthwise()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Ddepthwiseconv_float.h37 const int depth_multiplier = params.depth_multiplier; in DepthwiseConv() local
Ddepthwiseconv_uint8.h105 const int depth_multiplier = params.depth_multiplier; in Run() local
Dlegacy_reference_ops.h44 int pad_width, int pad_height, int depth_multiplier, in DepthwiseConv()
70 int pad_height, int depth_multiplier, in DepthwiseConv()
86 int pad_height, int depth_multiplier, float* output_data, in DepthwiseConv()
101 int pad_width, int pad_height, int depth_multiplier, in DepthwiseConv()
114 int pad_width, int pad_height, int depth_multiplier, in DepthwiseConv()
148 int pad_height, int depth_multiplier, in DepthwiseConv()
168 int pad_height, int depth_multiplier, int32 output_offset, in DepthwiseConv()
190 int pad_width, int pad_height, int depth_multiplier, in DepthwiseConv()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
Ddepthwise_conv.h36 const int depth_multiplier = params.depth_multiplier; in DepthwiseConvPerChannel() local
/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.h362 const dnn::FilterDescriptor& filter_descriptor, int depth_multiplier, in DoSeparableConvolve()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h353 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/contrib/tensor_forest/proto/
Dtensor_forest_params.proto94 float depth_multiplier = 4; field
/external/tensorflow/tensorflow/python/keras/layers/
Dconvolutional.py1236 depth_multiplier=1, argument
1439 depth_multiplier=1, argument
1608 depth_multiplier=1, argument
1740 depth_multiplier=1, argument
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h143 int pad_width, int pad_height, int depth_multiplier, in DepthwiseConv()
169 int pad_height, int depth_multiplier, in DepthwiseConv()
185 int pad_height, int depth_multiplier, float* output_data, in DepthwiseConv()
200 int pad_width, int pad_height, int depth_multiplier, in DepthwiseConv()
213 int pad_width, int pad_height, int depth_multiplier, in DepthwiseConv()
247 int pad_height, int depth_multiplier, in DepthwiseConv()
267 int pad_height, int depth_multiplier, int32 output_offset, in DepthwiseConv()
289 int pad_width, int pad_height, int depth_multiplier, in DepthwiseConv()
/external/tensorflow/tensorflow/lite/experimental/micro/kernels/portable_optimized/
Ddepthwise_conv.cc105 const int depth_multiplier = params.depth_multiplier; in DepthwiseConvOptimizedForFilterWidthEight() local
/external/tensorflow/tensorflow/lite/c/
Dbuiltin_op_data.h96 int depth_multiplier; member
/external/tensorflow/tensorflow/compiler/xla/
Dreference_util.cc123 const int64 depth_multiplier = depthwise_weights.planes(); in SeparableConvArray4D() local
/external/tensorflow/tensorflow/compiler/tests/
Drandomized_tests.cc1781 int depth_multiplier = random_int(generator()); in TEST_F() local
1804 int depth_multiplier = random_int(generator()); in TEST_F() local
1831 int depth_multiplier = random_int(generator()); in TEST_F() local
/external/tensorflow/tensorflow/contrib/layers/python/layers/
Dlayers.py2666 depth_multiplier=1, argument

12