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.cc46 PoolingOp(OpKernelConstruction* ctx, int num_spatial_dims, in PoolingOp()
144 int num_spatial_dims) { in XlaTensorFormat()
160 MaxPoolOp(OpKernelConstruction* ctx, int num_spatial_dims) in MaxPoolOp()
253 AvgPoolOp(OpKernelConstruction* ctx, int num_spatial_dims) in AvgPoolOp()
311 MaxPoolGradOp(OpKernelConstruction* ctx, int num_spatial_dims) in MaxPoolGradOp()
433 AvgPoolGradOp(OpKernelConstruction* ctx, int num_spatial_dims) in AvgPoolGradOp()
532 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()
164 explicit ConvBackpropFilterOp(OpKernelConstruction* ctx, int num_spatial_dims, in ConvBackpropFilterOp()
Dimage_resize_ops.cc90 int num_spatial_dims = in_size.size(); in ComputeResizeConvolutionParameters() local
234 const int num_spatial_dims, absl::Span<const int64> in_size, in ResizeUsingDilationAndConvolution()
380 const int num_spatial_dims, absl::Span<const int64> in_size, in ResizeUsingDilationAndConvolutionGradOp()
499 const int num_spatial_dims = 2; in GeneralCompile() local
678 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 StatusOr<ConvOpAttrs> ConvOpAttrs::Create(int num_spatial_dims, bool depthwise, 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.cc96 StringPiece label, int num_spatial_dims, const TensorShape& input_shape, in ConvBackpropComputeDimensionsV2()
160 Status ConvBackpropComputeDimensions(StringPiece label, int num_spatial_dims, in ConvBackpropComputeDimensions()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emission_utils.cc87 const int64_t num_spatial_dims = dnums.output_spatial_dimensions_size(); in PotentiallyImplementedAsEigenConvolution() local
Dconv_canonicalization.cc46 const int64_t num_spatial_dims = dnums.output_spatial_dimensions_size(); in Run() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dlegalize_hlo.cc111 const int num_spatial_dims = in matchAndRewrite() local
146 bool IsSamePadding(mhlo::ConvOp conv_op, int num_spatial_dims, in IsSamePadding()
196 int num_spatial_dims, RankedTensorType type, in GetReformatTypeAndPermutation()
221 int default_spatial_dim_start, int num_spatial_dims, in FormatToNHWC()
245 int input_channels, int num_spatial_dims, in CreateConvOp()
365 int num_spatial_dims = in IsSupportedConvOp() local
413 const int num_spatial_dims = conv_op.dimension_numbers() in matchAndRewrite() local
450 bool IsSamePadding(mhlo::ConvOp conv_op, int num_spatial_dims, in IsSamePadding()
504 int num_spatial_dims = in IsSupportedConvOp() 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/python/ops/
Dnn_ops.py680 def build_op(num_spatial_dims, padding): argument
865 def _with_space_to_batch_base_paddings(filter_shape, num_spatial_dims, argument
941 def _get_strides_and_dilation_rate(num_spatial_dims, strides, dilation_rate): argument
1157 num_spatial_dims=None): argument
1332 num_spatial_dims=None): argument
/external/tensorflow/tensorflow/compiler/mlir/xla/experimental/conv_emitter/
Dconv_emitter.cc290 int num_spatial_dims = output_spatial_indvars.size(); in CreateNaiveMlirConv() local
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.cc606 constexpr int num_spatial_dims = 2; in Conv2DShapeImpl() local
1132 constexpr int num_spatial_dims = 2; in AvgPoolShape() local
1575 constexpr int num_spatial_dims = 2; in MaxPoolShapeImpl() local
1714 constexpr int num_spatial_dims = 2; in MaxPoolV2Shape() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_a_m.cc1539 int num_spatial_dims = std::is_same<OpT, Conv2DOp>() ? 2 : 3; in Verify() local
1594 const int64_t num_spatial_dims = std::is_same<OpT, Conv2DOpAdaptor>() ? 2 : 3; in inferConvReturnTypes() local
1826 int num_spatial_dims = 2; in Verify() local
/external/tensorflow/tensorflow/core/ops/
Darray_ops.cc2493 constexpr int num_spatial_dims = 2; in __anon5ce23fde3702() local
2547 constexpr int num_spatial_dims = 2; in __anon5ce23fde3802() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference.cc1647 const int num_spatial_dims = dnums.input_spatial_dimensions_size(); in InferConvolveShape() local
Delemental_ir_emitter.cc3015 int num_spatial_dims = dnums.output_spatial_dimensions_size(); in EmitConvolution() local

12