Home
last modified time | relevance | path

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

1234

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A 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 …]
H A 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 …]
H A Deigen_spatial_convolutions_test.cc33 const int output_depth = 10; in TEST() local
78 const int output_depth = 10; in TEST() local
198 const int output_depth = 7; in TEST() local
250 const int output_depth = 7; in TEST() local
308 const int output_depth = 7; in TEST() local
364 const int output_depth = 7; in TEST() local
412 const int output_depth = 6; in TEST() local
460 const int output_depth = 7; in TEST() local
511 const int output_depth = 7; in TEST() local
562 const int output_depth = 7; in TEST() local
[all …]
H A 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
H A 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
H A 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
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
H A 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
H A 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
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/
H A 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
H A 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
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/
H A 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
H A 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
H A 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
H A Daveragepool_quantized_test.cc67 const int output_depth = input_depth; in CreateDataAndRunAveragePool() local
130 const int output_depth = input_depth; in CreateExtremalDataAndRunAveragePool() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/
H A DUpSampleNearest3d.cpp34 int64_t output_depth, in upsample_nearest3d_out_frame()
91 int64_t output_depth, in upsample_nearest3d_out_frame_nhwc()
149 int64_t output_depth = output_size[0]; in _upsample_nearest3d_quantized_cpu() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DConvolutionMM3d.cpp54 const int64_t output_depth = in compute_columns3d() local
200 const int64_t output_depth = in slow_conv3d_shape_check() local
284 int64_t output_depth, in slow_conv3d_update_output_frame()
419 const int64_t output_depth = in slow_conv3d_backward_out_cpu_template() local
608 const int64_t output_depth = in slow_conv3d_forward_out_cpu() local
H A DNaiveConvolutionTranspose3d.cpp134 const int64_t output_depth = (input_depth - 1) * stride_depth - in slow_conv_transpose3d_shape_check() local
269 const int64_t output_depth = (input_depth - 1) * stride_depth - in slow_conv_transpose3d_out_cpu_template() local
490 const int64_t output_depth = (input_depth - 1) * stride_depth - in slow_conv_transpose3d_backward_out_cpu_template() local
708 const int64_t output_depth = (input_depth - 1) * stride_depth - in slow_conv_transpose3d_acc_grad_parameters_cpu() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DNaiveConvolutionTranspose3d.cu134 int64_t output_depth = (input_depth - 1) * stride_depth - 2 * padding_depth + in slow_conv_transpose3d_shape_check() local
272 int64_t output_depth = (input_depth - 1) * stride_depth - 2 * padding_depth + in slow_conv_transpose3d_out_cuda_template() local
502 int64_t output_depth = (input_depth - 1) * stride_depth - 2 * padding_depth + in slow_conv_transpose3d_backward_out_cuda_template() local
727 int64_t output_depth = (input_depth - 1) * stride_depth - 2 * padding_depth + in slow_conv_transpose3d_acc_grad_parameters_cuda() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/sparse_ops/
H A 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
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
H A 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
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/nnapi/
H A 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
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/
H A 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
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DAdaptiveAvgPoolKernel.cpp429 int64_t output_depth = output_size[0]; in cpu_adaptive_avg_pool3d() local
494 int64_t output_depth = output_size[0]; in cpu_adaptive_avg_pool3d_channels_last() local
590 int64_t output_depth = output_size[0]; in cpu_adaptive_avg_pool3d_channels_last() local
697 int64_t output_depth = grad_output.size(-3); in cpu_adaptive_avg_pool3d_backward() local
756 int64_t output_depth = grad_output.size(2); in cpu_adaptive_avg_pool3d_backward_channels_last() local
H A DPaddingKernel.cpp149 int64_t output_depth = ndim == 3 ? p.oshape[ndim - 3] : 1; in cpu_padding() local
256 int64_t output_depth = ndim == 3 ? p.oshape[ndim - 3] : 1; in cpu_padding_channels_last() local
330 int64_t output_depth = ndim == 3 ? p.oshape[ndim - 3] : 1; in cpu_padding_backward() local
418 int64_t output_depth = ndim == 3 ? p.oshape[ndim - 3] : 1; in cpu_padding_backward_channels_last() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
H A 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

1234