| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | pooling_ops.cc | 48 PoolingOp(OpKernelConstruction* ctx, int num_spatial_dims, in PoolingOp()
146 int num_spatial_dims) { in XlaTensorFormat()
162 MaxPoolOp(OpKernelConstruction* ctx, int num_spatial_dims) in MaxPoolOp()
255 AvgPoolOp(OpKernelConstruction* ctx, int num_spatial_dims) in AvgPoolOp()
313 MaxPoolGradOp(OpKernelConstruction* ctx, int num_spatial_dims) in MaxPoolGradOp()
442 AvgPoolGradOp(OpKernelConstruction* ctx, int num_spatial_dims) in AvgPoolGradOp()
543 MaxPoolGradGradOp(OpKernelConstruction* ctx, int num_spatial_dims) in MaxPoolGradGradOp()
|
| H A D | conv_ops.cc | 45 explicit ConvOp(OpKernelConstruction* ctx, int num_spatial_dims, in ConvOp()
96 explicit ConvBackpropInputOp(OpKernelConstruction* ctx, int num_spatial_dims, in ConvBackpropInputOp()
164 explicit ConvBackpropFilterOp(OpKernelConstruction* ctx, int num_spatial_dims, in ConvBackpropFilterOp()
|
| H A D | image_resize_ops.cc | 95 int num_spatial_dims = in_size.size(); in ComputeResizeConvolutionParameters() local
239 const int num_spatial_dims, absl::Span<const int64_t> in_size, in ResizeUsingDilationAndConvolution()
385 const int num_spatial_dims, absl::Span<const int64_t> in_size, in ResizeUsingDilationAndConvolutionGradOp()
774 const int num_spatial_dims = 2; in Compile() local
|
| H A D | conv_op_helpers.cc | 66 int num_spatial_dims) { in TransposeFilterForGroupConvolutionBackpropInput()
137 StringPiece label, int num_spatial_dims, const xla::Shape& input_shape, in ConvBackpropComputeDimensionsV2XlaShapes()
160 StatusOr<ConvOpAttrs> ConvOpAttrs::Create(int num_spatial_dims, bool depthwise, in Create()
|
| H A D | conv_op_helpers.h | 50 int num_spatial_dims; member
|
| H A D | depthtospace_op.cc | 70 int num_spatial_dims = GetTensorSpatialDims(input_rank, data_format); in Compile() local
|
| H A D | extract_image_patches_op.cc | 50 const int num_spatial_dims = num_dims - 2; in Compile() local
|
| H A D | spacetodepth_op.cc | 70 int num_spatial_dims = GetTensorSpatialDims(input_rank, data_format); in Compile() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| H A D | pooling.cc | 35 const int num_spatial_dims = spatial_padding.size(); in AvgPoolDivideByCountWithGeneralPadding() local
91 int num_spatial_dims, absl::Span<const int64_t> stride, in MakeSpatialPaddingConfig()
161 const int num_spatial_dims = num_dims - 2; in AvgPool() local
177 const int num_spatial_dims = kernel_size.size() - 2; in MakeSpatialPadding() local
248 const int num_spatial_dims = num_dims - 2; in AvgPoolGrad() local
|
| H A D | pooling_test.cc | 25 TensorFormat MakeNCHWFormat(int num_spatial_dims) { in MakeNCHWFormat()
51 const int num_spatial_dims = spatial_dim_sizes.size(); in ExpandWithBatchAndFeatureDimensions() local
|
| H A D | pooling.h | 40 int num_spatial_dims() const { return spatial_dimensions_.size(); } in num_spatial_dims() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | conv_grad_shape_utils.cc | 96 StringPiece label, int num_spatial_dims, const TensorShape& input_shape, in ConvBackpropComputeDimensionsV2()
160 Status ConvBackpropComputeDimensions(StringPiece label, int num_spatial_dims, in ConvBackpropComputeDimensions()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| H A D | legalize_hlo.cc | 184 auto num_spatial_dims = in IsSupportedConvOp() local
438 const int num_spatial_dims = dnums.getInputSpatialDimensions().size(); in matchAndRewrite() local
472 bool IsSamePadding(mhlo::ConvolutionOp conv_op, int num_spatial_dims, in IsSamePadding()
517 int num_spatial_dims, RankedTensorType type, in GetReformatTypeAndPermutation()
542 int default_spatial_dim_start, int num_spatial_dims, in FormatToNHWC()
608 int input_channels, int num_spatial_dims, in CreateConvOp()
755 int num_spatial_dims = in matchAndRewrite() local
792 bool IsSamePadding(mhlo::ConvolutionOp conv_op, int num_spatial_dims, in IsSamePadding()
845 int num_spatial_dims = dnums.getInputSpatialDimensions().size(); in IsSupportedConvOp() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/util/ |
| H A D | tensor_format_test.cc | 115 GetTensorDimMap(const int num_spatial_dims, const TensorFormat format) { in GetTensorDimMap()
127 GetFilterDimMap(const int num_spatial_dims, in GetFilterDimMap()
|
| H A D | tensor_format.h | 144 inline int GetTensorDimsFromSpatialDims(int num_spatial_dims, in GetTensorDimsFromSpatialDims()
163 inline int GetFilterTensorDimsFromSpatialDims(int num_spatial_dims, in GetFilterTensorDimsFromSpatialDims()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | nn_ops.py | 713 def build_op(num_spatial_dims, padding): argument
905 def _with_space_to_batch_base_paddings(filter_shape, num_spatial_dims, argument
981 def _get_strides_and_dilation_rate(num_spatial_dims, strides, dilation_rate): argument
1206 num_spatial_dims=None): argument
1383 num_spatial_dims=None): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| H A D | ir_emission_utils.cc | 85 const int64_t num_spatial_dims = dnums.output_spatial_dimensions_size(); in PotentiallyImplementedAsEigenConvolution() local
|
| H A D | conv_canonicalization.cc | 48 const int64_t num_spatial_dims = dnums.output_spatial_dimensions_size(); in Run() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/nn_ops/ |
| H A D | atrous_convolution_test.py | 217 …def combined_op(converted_input, num_spatial_dims, padding_arg): # pylint: disable=unused-argument argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | common_shape_fns.cc | 606 constexpr int num_spatial_dims = 2; in Conv2DShapeImpl() local
1147 constexpr int num_spatial_dims = 2; in AvgPoolShape() local
1626 constexpr int num_spatial_dims = 2; in MaxPoolShapeImpl() local
1765 constexpr int num_spatial_dims = 2; in MaxPoolV2Shape() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/experimental/conv_emitter/ |
| H A D | conv_emitter.cc | 291 int num_spatial_dims = output_spatial_indvars.size(); in CreateNaiveMlirConv() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| H A D | tf_ops_a_m.cc | 1612 int num_spatial_dims = std::is_same<OpT, Conv2DOp>() ? 2 : 3; in Verify() local
1729 const int64_t num_spatial_dims = std::is_same<OpT, Conv2DOpAdaptor>() ? 2 : 3; in inferConvReturnTypeComponents() local
1926 int num_spatial_dims = 2; in verify() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/ops/ |
| H A D | array_ops.cc | 2513 constexpr int num_spatial_dims = 2; in __anon38bbb0e83702() local
2567 constexpr int num_spatial_dims = 2; in __anon38bbb0e83802() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | Convolution.cpp | 758 const int64_t num_spatial_dims, in batchify()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | shape_inference.cc | 1657 const int num_spatial_dims = dnums.input_spatial_dimensions_size(); in InferConvolveShape() local
|