| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | ConvolutionMM3d.cpp | 43 const int64_t stride_depth = stride[0]; in compute_columns3d() local
118 int64_t stride_depth, in slow_conv3d_shape_check()
272 int64_t stride_depth, in slow_conv3d_update_output_frame()
320 int64_t stride_depth, in slow_conv3d_backward_update_grad_input_frame()
385 const int64_t stride_depth = stride[0]; in slow_conv3d_backward_out_cpu_template() local
509 const int64_t stride_depth = stride[0]; in slow_conv3d_backward_parameters_out_cpu_template() local
570 const int64_t stride_depth = stride[0]; in slow_conv3d_forward_out_cpu() local
|
| H A D | NaiveConvolutionTranspose3d.cpp | 36 int stride_depth, in slow_conv_transpose3d_shape_check()
221 int64_t stride_depth = stride[0]; in slow_conv_transpose3d_out_cpu_template() local
435 int64_t stride_depth = stride[0]; in slow_conv_transpose3d_backward_out_cpu_template() local
640 int64_t stride_depth = stride[0]; in slow_conv_transpose3d_acc_grad_parameters_cpu() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | NaiveConvolutionTranspose3d.cu | 36 int stride_depth, in slow_conv_transpose3d_shape_check()
217 int64_t stride_depth = stride[0]; in slow_conv_transpose3d_out_cuda_template() local
441 int64_t stride_depth = stride[0]; in slow_conv_transpose3d_backward_out_cuda_template() local
649 int64_t stride_depth = stride[0]; in slow_conv_transpose3d_acc_grad_parameters_cuda() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | conv3d_test.cc | 34 Padding padding = Padding_VALID, int32_t stride_depth = 1, in Conv3dOpModel()
54 int32_t stride_depth = 1, int32_t stride_width = 1, in Conv3dOpModel()
|
| H A D | conv3d_transpose_test.cc | 41 int32_t stride_depth = 1, int32_t stride_width = 1, in Conv3dTransposeOpModel()
72 Padding padding = Padding_VALID, int32_t stride_depth = 1, in Conv3dTransposeOpModel()
|
| H A D | padding.h | 85 int stride_height, int stride_width, int stride_depth, in ComputePadding3DValues()
|
| H A D | pooling3d_test.cc | 77 std::vector<uint8_t> CreateCustomOptions(int stride_depth, int stride_height, in CreateCustomOptions()
|
| H A D | pooling3d.cc | 43 int stride_depth; member
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/ |
| H A D | indirection.c | 37 const size_t stride_depth = op->stride_depth; in pytorch_qnnp_indirection_init_conv3d() local
241 const size_t stride_depth = op->stride_depth; in pytorch_qnnp_indirection_init_dwconv() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| H A D | legalize_tf.cc | 335 IntegerAttr stride_depth, stride_height, stride_width; in matchAndRewrite() local
375 IntegerAttr stride_depth, stride_height, stride_width; in matchAndRewrite() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| H A D | im2col_utils.h | 290 int stride_depth, int stride_height, int stride_width, // Stride params. in ExtractPatchIntoBufferColumn3D()
379 const int stride_depth = params.stride_depth; in Im2col3D() local
|
| H A D | optimized_ops.h | 8056 const int stride_depth = params.stride_depth; in Conv3D() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| H A D | conv3d_transpose.h | 32 const int stride_depth = params.stride_depth; in Conv3DTranspose() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/qnnpack/ |
| H A D | operator.h | 67 uint32_t stride_depth; member
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | common_shape_fns.cc | 1619 int32_t stride_depth = GetTensorDim(strides, data_format, 'C'); in MaxPoolShapeImpl() local
1758 int32_t stride_depth = GetTensorDim(strides, data_format, 'C'); in MaxPoolV2Shape() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/c/ |
| H A D | builtin_op_data.h | 93 int stride_depth; member
|
| /aosp_15_r20/external/mesa3d/include/tensorflow/lite/core/c/ |
| H A D | builtin_op_data.h | 98 int stride_depth; member
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/ |
| H A D | types.h | 724 int stride_depth; member
|