Home
last modified time | relevance | path

Searched defs:stride_depth (Results 1 – 18 of 18) sorted by relevance

/external/pytorch/aten/src/ATen/native/
DConvolutionMM3d.cpp43 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
DNaiveConvolutionTranspose3d.cpp36 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
/external/tensorflow/tensorflow/lite/kernels/
Dconv3d_test.cc34 Padding padding = Padding_VALID, int32_t stride_depth = 1, in Conv3dOpModel()
54 int32_t stride_depth = 1, int32_t stride_width = 1, in Conv3dOpModel()
Dconv3d_transpose_test.cc41 int32_t stride_depth = 1, int32_t stride_width = 1, in Conv3dTransposeOpModel()
72 Padding padding = Padding_VALID, int32_t stride_depth = 1, in Conv3dTransposeOpModel()
Dpadding.h85 int stride_height, int stride_width, int stride_depth, in ComputePadding3DValues()
Dpooling3d_test.cc77 std::vector<uint8_t> CreateCustomOptions(int stride_depth, int stride_height, in CreateCustomOptions()
Dpooling3d.cc43 int stride_depth; member
/external/pytorch/aten/src/ATen/native/cuda/
DNaiveConvolutionTranspose3d.cu36 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
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dconv3d_transpose.h32 const int stride_depth = params.stride_depth; in Conv3DTranspose() local
/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/
Dindirection.c37 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
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dlegalize_tf.cc335 IntegerAttr stride_depth, stride_height, stride_width; in matchAndRewrite() local
375 IntegerAttr stride_depth, stride_height, stride_width; in matchAndRewrite() local
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dim2col_utils.h290 int stride_depth, int stride_height, int stride_width, // Stride params. in ExtractPatchIntoBufferColumn3D()
379 const int stride_depth = params.stride_depth; in Im2col3D() local
Doptimized_ops.h8056 const int stride_depth = params.stride_depth; in Conv3D() local
/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/qnnpack/
Doperator.h67 uint32_t stride_depth; member
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.cc1619 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
/external/tensorflow/tensorflow/lite/c/
Dbuiltin_op_data.h93 int stride_depth; member
/external/mesa3d/include/tensorflow/lite/core/c/
Dbuiltin_op_data.h98 int stride_depth; member
/external/tensorflow/tensorflow/lite/kernels/internal/
Dtypes.h724 int stride_depth; member