Home
last modified time | relevance | path

Searched defs:num_spatial_dims (Results 1 – 25 of 28) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dpooling_ops.cc43 PoolingOp(OpKernelConstruction* ctx, int num_spatial_dims, in PoolingOp()
138 int num_spatial_dims) { in XlaTensorFormat()
154 MaxPoolOp(OpKernelConstruction* ctx, int num_spatial_dims) in MaxPoolOp()
212 AvgPoolOp(OpKernelConstruction* ctx, int num_spatial_dims) in AvgPoolOp()
275 MaxPoolGradOp(OpKernelConstruction* ctx, int num_spatial_dims) in MaxPoolGradOp()
399 AvgPoolGradOp(OpKernelConstruction* ctx, int num_spatial_dims) in AvgPoolGradOp()
495 MaxPoolGradGradOp(OpKernelConstruction* ctx, int num_spatial_dims) in MaxPoolGradGradOp()
Dconv_ops.cc45 explicit ConvOp(OpKernelConstruction* ctx, int num_spatial_dims, in ConvOp()
96 explicit ConvBackpropInputOp(OpKernelConstruction* ctx, int num_spatial_dims, in ConvBackpropInputOp()
162 explicit ConvBackpropFilterOp(OpKernelConstruction* ctx, int num_spatial_dims, in ConvBackpropFilterOp()
Dimage_resize_ops.cc90 int num_spatial_dims = in_size.size(); in ComputeResizeConvolutionParameters() local
233 const int num_spatial_dims, absl::Span<const int64> in_size, in ResizeUsingDilationAndConvolution()
379 const int num_spatial_dims, absl::Span<const int64> in_size, in ResizeUsingDilationAndConvolutionGradOp()
498 const int num_spatial_dims = 2; in GeneralCompile() local
676 const int num_spatial_dims = 2; in Compile() local
Dconv_op_helpers.h50 int num_spatial_dims; member
Dconv_op_helpers.cc66 int num_spatial_dims) { in TransposeFilterForGroupConvolutionBackpropInput()
137 StringPiece label, int num_spatial_dims, const xla::Shape& input_shape, in ConvBackpropComputeDimensionsV2XlaShapes()
160 xla::StatusOr<ConvOpAttrs> ConvOpAttrs::Create(int num_spatial_dims, in Create()
Ddepthtospace_op.cc70 int num_spatial_dims = GetTensorSpatialDims(input_rank, data_format); in Compile() local
Dextract_image_patches_op.cc50 const int num_spatial_dims = num_dims - 2; in Compile() local
Dspacetodepth_op.cc70 int num_spatial_dims = GetTensorSpatialDims(input_rank, data_format); in Compile() local
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dpooling.cc35 const int num_spatial_dims = spatial_padding.size(); in AvgPoolDivideByCountWithGeneralPadding() local
91 int num_spatial_dims, absl::Span<const int64> 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
Dpooling.h40 int num_spatial_dims() const { return spatial_dimensions_.size(); } in num_spatial_dims() function
Dpooling_test.cc25 TensorFormat MakeNCHWFormat(int num_spatial_dims) { in MakeNCHWFormat()
51 const int num_spatial_dims = spatial_dim_sizes.size(); in ExpandWithBatchAndFeatureDimensions() local
/external/tensorflow/tensorflow/core/kernels/
Dconv_grad_shape_utils.cc95 StringPiece label, int num_spatial_dims, const TensorShape& input_shape, in ConvBackpropComputeDimensionsV2()
155 Status ConvBackpropComputeDimensions(StringPiece label, int num_spatial_dims, in ConvBackpropComputeDimensions()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emission_utils.cc87 const int64 num_spatial_dims = dnums.output_spatial_dimensions_size(); in PotentiallyImplementedAsEigenConvolution() local
Dconv_canonicalization.cc46 const int64 num_spatial_dims = dnums.output_spatial_dimensions_size(); in Run() local
/external/tensorflow/tensorflow/core/util/
Dtensor_format_test.cc115 GetTensorDimMap(const int num_spatial_dims, const TensorFormat format) { in GetTensorDimMap()
127 GetFilterDimMap(const int num_spatial_dims, in GetFilterDimMap()
Dtensor_format.h144 inline int GetTensorDimsFromSpatialDims(int num_spatial_dims, in GetTensorDimsFromSpatialDims()
163 inline int GetFilterTensorDimsFromSpatialDims(int num_spatial_dims, in GetFilterTensorDimsFromSpatialDims()
/external/tensorflow/tensorflow/python/kernel_tests/
Datrous_convolution_test.py221 …def combined_op(converted_input, num_spatial_dims, padding_arg): # pylint: disable=unused-argument argument
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dlegalize_hlo.cc120 const int num_spatial_dims = conv_op.dimension_numbers() in matchAndRewrite() local
136 bool IsSamePadding(mhlo::ConvOp conv_op, int num_spatial_dims, in IsSamePadding()
218 int num_spatial_dims = in IsSupportedConvOp() local
/external/tensorflow/tensorflow/python/ops/
Dnn_ops.py560 def build_op(num_spatial_dims, padding): argument
745 def _with_space_to_batch_base_paddings(filter_shape, num_spatial_dims, argument
821 def _get_strides_and_dilation_rate(num_spatial_dims, strides, dilation_rate): argument
1037 num_spatial_dims=None): argument
1212 num_spatial_dims=None): argument
/external/tensorflow/tensorflow/compiler/mlir/xla/experimental/conv_emitter/
Dconv_emitter.cc289 int num_spatial_dims = output_spatial_indvars.size(); in CreateNaiveMlirConv() local
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.cc598 constexpr int num_spatial_dims = 2; in Conv2DShapeImpl() local
1092 constexpr int num_spatial_dims = 2; in AvgPoolShape() local
1535 constexpr int num_spatial_dims = 2; in MaxPoolShapeImpl() local
1674 constexpr int num_spatial_dims = 2; in MaxPoolV2Shape() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_a_m.cc1516 int num_spatial_dims = std::is_same<OpT, Conv2DOp>() ? 2 : 3; in Verify() local
1571 const int64_t num_spatial_dims = std::is_same<OpT, Conv2DOpAdaptor>() ? 2 : 3; in inferConvReturnTypes() local
1803 int num_spatial_dims = 2; in Verify() local
/external/tensorflow/tensorflow/core/ops/
Darray_ops.cc2480 constexpr int num_spatial_dims = 2; in __anon42d741193702() local
2534 constexpr int num_spatial_dims = 2; in __anon42d741193802() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference.cc1658 const int num_spatial_dims = dnums.input_spatial_dimensions_size(); in InferConvolveShape() local
Delemental_ir_emitter.cc2830 int num_spatial_dims = dnums.output_spatial_dimensions_size(); in EmitConvolution() local

12