Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/python/keras/applications/
Dmobilenet.py87 depth_multiplier=1, argument
367 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()
190 int filter_width, int filter_height, int depth_multiplier, in GenerateValidShapeConfigurations()
228 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
620 int depth_multiplier, std::int32_t output_offset, in TestOneDepthwiseConvWithGivenOutputShift()
757 int depth_multiplier, std::int32_t output_offset, in TestOneDepthwiseConvBisectOutputShift()
816 int depth_multiplier, std::int32_t output_offset, in TestOneDepthwiseConv()
833 int depth_multiplier, int stride, in TryTestDepthwiseConv()
963 const int depth_multiplier = ExponentialRandomPositiveInt(0.8f, 6, 50); in TryTestOneDepthwiseConv() local
985 const int depth_multiplier = 1; in TryTestOneDepthwiseConv3x3Filter() local
1037 const int depth_multiplier = in TryTestOneNeonDot3x3() local
Ddepthwiseconv_per_channel_quantized_16x8_test.cc51 const int depth_multiplier = params.depth_multiplier; in PickOutputMultiplier() local
142 int filter_width, int filter_height, int depth_multiplier, in GenerateValidShapeConfigurations()
186 const int depth_multiplier = 1; in TryTestOneDepthwiseConv3x3Filter() local
Ddepthwiseconv_float_test.cc80 const int depth_multiplier = ExponentialRandomPositiveInt(0.8f, 6, 50); in TryTestOneDepthwiseConv() local
/external/XNNPACK/src/subgraph/
Ddepthwise-convolution-2d.c28 uint32_t depth_multiplier, in xnn_define_depthwise_convolution_2d()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
Ddepthwise_conv.h37 const int depth_multiplier = params.depth_multiplier; in DepthwiseConvPerChannel() local
136 const int depth_multiplier = params.depth_multiplier; in DepthwiseConvPerChannel() local
221 const int depth_multiplier = params.depth_multiplier; in DepthwiseConvHybridPerChannel() local
/external/tensorflow/tensorflow/core/kernels/
Ddepthwise_conv_op.h33 int depth_multiplier; member
Ddepthwise_conv_op.cc350 const int32 depth_multiplier = filter.dim_size(3); in Compute() local
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Ddepthwiseconv_uint8.h104 const int depth_multiplier = params.depth_multiplier; in Run() local
190 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.h47 int pad_width, int pad_height, int depth_multiplier, in DepthwiseConv()
73 int pad_height, int depth_multiplier, in DepthwiseConv()
89 int pad_height, int depth_multiplier, float* output_data, in DepthwiseConv()
104 int pad_width, int pad_height, int depth_multiplier, in DepthwiseConv()
117 int pad_width, int pad_height, int depth_multiplier, in DepthwiseConv()
151 int pad_height, int depth_multiplier, in DepthwiseConv()
171 int pad_height, int depth_multiplier, int32 output_offset, in DepthwiseConv()
193 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.h381 const dnn::FilterDescriptor& filter_descriptor, int depth_multiplier, in DoSeparableConvolve()
/external/tensorflow/tensorflow/python/keras/legacy_tf_layers/
Dconvolutional.py723 depth_multiplier=1, argument
838 depth_multiplier=1, argument
887 depth_multiplier=1, argument
1010 depth_multiplier=1, argument
/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Ddepthwise_conv_2d_tester.h51 inline DepthwiseConv2DTester& DepthMultiplier(int32_t depth_multiplier) { in DepthMultiplier()
/external/XNNPACK/test/
Dsubgraph-tester.h103 uint32_t depth_multiplier, size_t input_channels, uint32_t input_id, in add_depthwise_conv()
/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/stream_executor/rocm/
Drocm_dnn.h405 const dnn::FilterDescriptor& filter_descriptor, int depth_multiplier, in DoSeparableConvolve()
/external/tensorflow/tensorflow/lite/micro/kernels/vexriscv/
Ddepthwise_conv.cc50 const int depth_multiplier = params.depth_multiplier; in DepthwiseConvPerChannel() local
172 const int depth_multiplier = params.depth_multiplier; in DepthwiseConv() local
/external/tensorflow/tensorflow/python/keras/layers/
Dconvolutional.py1771 depth_multiplier=1, argument
2004 depth_multiplier=1, argument
2194 depth_multiplier=1, argument
2349 depth_multiplier=1, argument
/external/llvm-project/mlir/lib/Dialect/Linalg/EDSC/
DBuilders.cpp247 Value vI, Value vW, Value vO, int depth_multiplier, ArrayRef<int> strides, in linalg_generic_dilated_conv_nhwc()
/external/tensorflow/tensorflow/lite/kernels/
Ddepthwise_conv.cc285 int16* depth_multiplier) { in ComputeDepthMultiplier()
/external/tensorflow/tensorflow/lite/micro/kernels/xtensa/
Ddepthwise_conv.cc80 const int depth_multiplier = params.depth_multiplier; in DepthwiseConvPerChannel() local
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dmodel_builder.cc433 const int depth_multiplier = tf_options->depth_multiplier; in IsSupported() local
494 const int depth_multiplier = tf_options->depth_multiplier; in Parse() local
512 const TfLiteTensor* output, int depth_multiplier, in TransposeWeights()

12