Home
last modified time | relevance | path

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

/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/tensorflow/tensorflow/lite/kernels/internal/reference/
Dconv3d_transpose.h32 const int stride_depth = params.stride_depth; in Conv3DTranspose() 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/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/tensorflow/tensorflow/lite/kernels/internal/
Dtypes.h724 int stride_depth; member