Home
last modified time | relevance | path

Searched defs:output_depth (Results 1 – 25 of 65) sorted by relevance

123

/external/tensorflow/tensorflow/core/kernels/
Deigen_backward_spatial_convolutions_test.cc34 const int output_depth = 5; in TEST() local
83 const int output_depth = 5; in TEST() local
135 const int output_depth = 3; in TEST() local
183 const int output_depth = 3; in TEST() local
235 const int output_depth = 3; in TEST() local
290 const int output_depth = 3; in TEST() local
342 const int input_cols, const int output_depth) { in test_batched_strided_spatial_convolution_backward_input_valid()
402 int output_depth = 1; in TEST() local
418 const int input_cols, const int output_depth) { in test_batched_strided_spatial_convolution_backward_input_valid_row_major()
479 int output_depth = 1; in TEST() local
[all …]
Deigen_backward_cuboid_convolutions_test.cc41 const int output_depth = 5; in TEST() local
104 const int output_depth = 5; in TEST() local
167 const int output_depth = 5; in TEST() local
238 const int output_depth = 5; in TEST() local
310 const int output_depth = 5; in TEST() local
377 const int output_depth = 5; in TEST() local
437 const int output_depth = 5; in TEST() local
503 const int output_depth = 5; in TEST() local
570 const int output_depth = 3; in TEST() local
637 const int output_depth = 3; in TEST() local
[all …]
Ddepthtospace_op_gpu.cu.cc38 const int output_width, const int output_depth, in D2S_NHWC()
156 const int output_depth = output.dimension(3); in operator ()() local
184 const int output_depth = output.dimension(1); in operator ()() local
Dspacetodepth_op_gpu.cu.cc37 const int output_width, const int output_depth, in S2D_NHWC()
153 const int output_depth = output.dimension(3); in operator ()() local
179 const int output_depth = output.dimension(1); in operator ()() local
Ddepthtospace_op.cc100 const int output_depth = input_depth / block_size_sq; in Compute() local
155 const int output_depth = output.dimension(3); in operator ()() local
/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
Dfully_connected.h49 const int output_depth = output_shape.Dims(1); in FullyConnectedPerChannel() local
90 const int output_depth = output_shape.Dims(output_dim_count - 1); in FullyConnectedPerChannel() local
133 const int output_depth = output_shape.Dims(output_dim_count - 1); in FullyConnected() local
175 const int output_depth = output_shape.Dims(output_dim_count - 1); in FullyConnected() local
Ddepthwise_conv.h52 const int output_depth = MatchingDim(filter_shape, 3, output_shape, 3); in DepthwiseConvPerChannel() local
149 const int output_depth = MatchingDim(filter_shape, 3, output_shape, 3); in DepthwiseConvPerChannel() local
232 const int output_depth = MatchingDim(filter_shape, 3, output_shape, 3); in DepthwiseConvHybridPerChannel() local
Dmean.h42 const int output_depth = output_shape.Dims(3); in Mean() local
Dconv.h54 const int output_depth = MatchingDim(filter_shape, 0, output_shape, 3); in ConvPerChannel() local
165 const int output_depth = MatchingDim(filter_shape, 0, output_shape, 3); in ConvPerChannel() local
Dtranspose_conv.h46 const int output_depth = MatchingDim(filter_shape, 0, output_shape, 3); in TransposeConv() local
146 const int output_depth = MatchingDim(filter_shape, 0, output_shape, 3); in TransposeConv() local
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dquant_lstm_sup.cc59 const auto output_depth = OutputDepth(weight_dims); in SetWeightSubmatrixDims() local
91 const auto output_depth = OutputDepth(weight_dims); in DecomposeQuantLstmWeightsTensor() local
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dfully_connected.h45 const int output_depth = MatchingDim(weights_shape, weights_dims_count - 2, in FullyConnected() local
90 const int output_depth = MatchingDim(filter_shape, filter_dim_count - 2, in FullyConnected() local
137 const int output_depth = MatchingDim(filter_shape, filter_dim_count - 2, in FullyConnected() local
190 const int output_depth = MatchingDim(weights_shape, weights_dim_count - 2, in ShuffledFullyConnected() local
Dconv.h49 const int output_depth = MatchingDim(filter_shape, 0, output_shape, 3); in Conv() local
139 const int output_depth = MatchingDim(filter_shape, 0, output_shape, 3); in Conv() local
226 const int output_depth = MatchingDim(filter_shape, 0, output_shape, 3); in HybridConvPerChannel() local
Ddepth_to_space.h41 const int output_depth = output_shape.Dims(3); in DepthToSpace() local
Dspace_to_depth.h43 const int output_depth = output_shape.Dims(3); in SpaceToDepth() local
Dtranspose_conv.h45 const int output_depth = MatchingDim(filter_shape, 0, output_shape, 3); in TransposeConv() local
135 const int output_depth = MatchingDim(filter_shape, 0, output_shape, 3); in TransposeConv() local
Ddepthwiseconv_uint8.h140 const int output_depth = MatchingDim(filter_shape, 3, output_shape, 3); in Run() local
227 const int output_depth = MatchingDim(filter_shape, 3, output_shape, 3); in RunPerChannel() local
/external/tensorflow/tensorflow/lite/kernels/internal/
Dconv_per_channel_quantized_16x8_test.cc60 const int output_depth = output_shape.Dims(3); in PickOutputMultiplier() local
116 int output_activation_max, int output_depth, in PickReasonableMultiplier()
155 const int output_depth = 8 * ExponentialRandomPositiveInt(0.9f, 10, 50); in GenerateValidShapeConfigurations() local
212 const int output_depth = output_shape_inference.Dims(3); in TryTestOneConvFilter() local
Daveragepool_quantized_test.cc67 const int output_depth = input_depth; in CreateDataAndRunAveragePool() local
130 const int output_depth = input_depth; in CreateExtremalDataAndRunAveragePool() local
Ddepthwiseconv_per_channel_quantized_16x8_test.cc115 int output_activation_max, int output_depth, in PickReasonableMultiplier()
155 const int output_depth = input_depth * depth_multiplier; in GenerateValidShapeConfigurations() local
211 const int output_depth = output_shape_inference.Dims(3); in TryTestOneDepthwiseConv3x3Filter() local
Ddepthwiseconv_per_channel_quantized_test.cc116 int output_activation_max, int output_depth, in PickReasonableMultiplier()
203 const int output_depth = input_depth * depth_multiplier; in GenerateValidShapeConfigurations() local
256 const int output_depth = output_shape_inference.Dims(3); in TryTestOneDepthwiseConv3x3Filter() local
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dfuse_binary_into_following_affine.cc79 const int output_depth = weights_shape.dims(0); in FuseAddOrSubParamsIntoFollowingAffine() local
95 const int output_depth = in FuseAddOrSubParamsIntoFollowingAffine() local
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/sparse_ops/
Dfully_connected.h48 const int output_depth = MatchingDim(weights_shape, weights_dims_count - 2, in FullyConnectedSparseWeight() local
95 const int output_depth = MatchingDim(weights_shape, weights_dims_count - 2, in FullyConnectedSparseWeight1x16Impl() local
133 const int output_depth = MatchingDim(weights_shape, weights_dims_count - 2, in FullyConnectedSparseWeight1x4Impl() local
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
Ddepthwise_conv_hybrid.h35 inline void DepthwiseConvInitAccBuffer(int num_output_pixels, int output_depth, in DepthwiseConvInitAccBuffer()
61 const int output_depth = MatchingDim(filter_shape, 3, output_shape, 3); in DepthwiseConvHybridGeneral() local
278 const int output_depth = MatchingDim(filter_shape, 3, output_shape, 3); in DepthwiseConvHybridWithRounding() local
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dim2col_utils.h229 const int output_depth = output_shape.Dims(3); in Im2col() local
266 const int output_depth = output_shape.Dims(3); in Im2col() local
394 const int output_depth = im2col_shape.Dims(1); in Im2col3D() local
436 const int output_depth = im2col_shape.Dims(1); in DilatedIm2col3D() local

123