| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | conv_ops.cc | 44 explicit ConvOp(OpKernelConstruction* ctx, int num_spatial_dims, in ConvOp()
91 explicit ConvBackpropInputOp(OpKernelConstruction* ctx, int num_spatial_dims, in ConvBackpropInputOp()
149 explicit ConvBackpropFilterOp(OpKernelConstruction* ctx, int num_spatial_dims, in ConvBackpropFilterOp()
|
| D | pooling_ops.cc | 41 PoolingOp(OpKernelConstruction* ctx, int num_spatial_dims, in PoolingOp()
136 int num_spatial_dims) { in XlaTensorFormat()
152 MaxPoolOp(OpKernelConstruction* ctx, int num_spatial_dims) in MaxPoolOp()
204 AvgPoolOp(OpKernelConstruction* ctx, int num_spatial_dims) in AvgPoolOp()
268 MaxPoolGradOp(OpKernelConstruction* ctx, int num_spatial_dims) in MaxPoolGradOp()
378 AvgPoolGradOp(OpKernelConstruction* ctx, int num_spatial_dims) in AvgPoolGradOp()
475 MaxPoolGradGradOp(OpKernelConstruction* ctx, int num_spatial_dims) in MaxPoolGradGradOp()
|
| D | image_resize_ops.cc | 85 int num_spatial_dims = in_size.size(); in ComputeResizeConvolutionParameters() local
228 const int num_spatial_dims, absl::Span<const int64> in_size, in ResizeUsingDilationAndConvolution()
356 const int num_spatial_dims, absl::Span<const int64> in_size, in ResizeUsingDilationAndConvolutionGradOp()
475 const int num_spatial_dims = 2; in GeneralCompile() local
640 const int num_spatial_dims = 2; in Compile() local
|
| D | conv_op_helpers.h | 46 int num_spatial_dims; member
|
| D | depthtospace_op.cc | 70 int num_spatial_dims = GetTensorSpatialDims(input_rank, data_format); in Compile() local
|
| D | extract_image_patches_op.cc | 43 const int num_spatial_dims = num_dims - 2; in Compile() local
|
| D | spacetodepth_op.cc | 70 int num_spatial_dims = GetTensorSpatialDims(input_rank, data_format); in Compile() local
|
| D | conv_op_helpers.cc | 203 StringPiece label, int num_spatial_dims, const xla::Shape& input_shape, in ConvBackpropComputeDimensionsV2XlaShapes()
222 xla::StatusOr<ConvOpAttrs> ConvOpAttrs::Create(int num_spatial_dims, in Create()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | pooling.cc | 35 const int num_spatial_dims = spatial_padding.size(); in AvgPoolDivideByCountWithGeneralPadding() local
91 int num_spatial_dims, absl::Span<const int64> stride, in MakeSpatialPaddingConfig()
160 const int num_spatial_dims = num_dims - 2; in AvgPool() local
176 const int num_spatial_dims = kernel_size.size() - 2; in MakeSpatialPadding() local
245 const int num_spatial_dims = num_dims - 2; in AvgPoolGrad() local
|
| D | pooling.h | 40 int num_spatial_dims() const { return spatial_dimensions_.size(); } in num_spatial_dims() function
|
| 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
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_grad_ops.cc | 99 StringPiece label, int num_spatial_dims, const TensorShape& input_shape, in ConvBackpropComputeDimensionsV2()
159 Status ConvBackpropComputeDimensions(StringPiece label, int num_spatial_dims, in ConvBackpropComputeDimensions()
|
| /external/tensorflow/tensorflow/contrib/fused_conv/ops/ |
| D | fused_conv2d_bias_activation_op.cc | 63 constexpr int num_spatial_dims = 2; in __anon4e6e467d0102() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emission_utils.cc | 87 const int64 num_spatial_dims = dnums.output_spatial_dimensions_size(); in PotentiallyImplementedAsEigenConvolution() local
|
| D | conv_canonicalization.cc | 45 const int64 num_spatial_dims = dnums.output_spatial_dimensions_size(); in Run() local
|
| /external/tensorflow/tensorflow/core/util/ |
| 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()
|
| D | tensor_format.h | 136 inline int GetTensorDimsFromSpatialDims(int num_spatial_dims, in GetTensorDimsFromSpatialDims()
152 inline int GetFilterTensorDimsFromSpatialDims(int num_spatial_dims, in GetFilterTensorDimsFromSpatialDims()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | atrous_convolution_test.py | 221 …def combined_op(converted_input, num_spatial_dims, padding_arg): # pylint: disable=unused-argument argument
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 431 constexpr int num_spatial_dims = 2; in Conv2DShapeImpl() local
758 constexpr int num_spatial_dims = 2; in AvgPoolShape() local
910 constexpr int num_spatial_dims = 2; in MaxPoolShape() local
1012 constexpr int num_spatial_dims = 2; in MaxPoolV2Shape() local
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_ops.py | 467 def build_op(num_spatial_dims, padding): argument
637 def _with_space_to_batch_base_paddings(filter_shape, num_spatial_dims, argument
716 def _get_strides_and_dilation_rate(num_spatial_dims, strides, dilation_rate): argument
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_ops.cc | 2323 constexpr int num_spatial_dims = 2; in __anon7c94107b3702() local
2377 constexpr int num_spatial_dims = 2; in __anon7c94107b3802() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference.cc | 1618 const int num_spatial_dims = dnums.input_spatial_dimensions_size(); in InferConvolveShape() local
|
| D | hlo_evaluator_typed_visitor.h | 1025 const int64 num_spatial_dims = dnums.output_spatial_dimensions_size(); in HandleConvolution() local
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 1070 int num_spatial_dims = num_dims - 2; in VerifyConvolution() local
2646 XlaBuilder::CreateDefaultConvDimensionNumbers(int num_spatial_dims) { in CreateDefaultConvDimensionNumbers()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_client.py | 1659 def _GetConvDimensionNumbers(self, num_spatial_dims): argument
|