Searched refs:stride_depth (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | padding.h | 81 int stride_height, int stride_width, int stride_depth, in ComputePadding3DValues() argument
90 *out_depth = ComputeOutSize(padding, in_depth, filter_depth, stride_depth, in ComputePadding3DValues()
96 ComputePaddingWithOffset(stride_depth, dilation_rate_depth, in_depth, in ComputePadding3DValues()
|
| D | conv3d_test.cc | 34 Padding padding = Padding_VALID, int32_t stride_depth = 1, in Conv3dOpModel() argument
45 CreateConv3DOptions(builder_, padding, stride_depth, stride_width, in Conv3dOpModel()
54 int32_t stride_depth = 1, int32_t stride_width = 1, in Conv3dOpModel() argument
64 CreateConv3DOptions(builder_, padding, stride_depth, stride_width, in Conv3dOpModel()
|
| D | conv3d.cc | 96 params->stride_height, params->stride_width, params->stride_depth, in Prepare()
129 runtime_params.stride_depth = params->stride_depth; in Eval()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | conv3d.h | 56 const int in_d_origin = (out_d * params.stride_depth) - pad_depth; in Conv3D()
|
| /external/tensorflow/tensorflow/lite/c/ |
| D | builtin_op_data.h | 94 int stride_depth; member
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | legalize_tf.cc | 348 IntegerAttr stride_depth, stride_height, stride_width; in matchAndRewrite() local
349 if (!TFIntListIs1XYZ1(op, "strides", &stride_depth, &stride_height, in matchAndRewrite()
377 stride_depth, stride_height, stride_width); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 1528 int32 stride_depth = GetTensorDim(strides, data_format, 'C'); in MaxPoolShapeImpl() local
1579 c, in_depth_dim, kernel_depth, /*dilation_rate=*/1, stride_depth, padding, in MaxPoolShapeImpl()
1667 int32 stride_depth = GetTensorDim(strides, data_format, 'C'); in MaxPoolV2Shape() local
1694 c, in_depth_dim, kernel_depth, stride_depth, padding, &output_depth)); in MaxPoolV2Shape()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | types.h | 879 int stride_depth; member
|
| /external/tensorflow/tensorflow/lite/core/api/ |
| D | flatbuffer_conversions.cc | 783 params->stride_depth = conv3d_params->stride_d(); in ParseOpDataTfLite()
|