Searched defs:spatial_dims (Results 1 – 10 of 10) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | batch_to_space_nd_test.cc | 74 int spatial_dims = static_cast<int>(block_shape.size()); in BatchToSpaceNDOpConstModel() local
104 int spatial_dims = static_cast<int>(input_shape.size()) - 2; in BatchToSpaceNDOpDynamicModel() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | conv_grad_shape_utils.h | 47 gtl::InlinedVector<ConvBackpropSpatialDimension, 3> spatial_dims; member
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | nn_ops.py | 575 spatial_dims=None, argument
752 spatial_dims=None, argument
922 def _with_space_to_batch_adjust(orig, fill_value, spatial_dims): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/experimental/conv_emitter/ |
| H A D | conv_emitter.cc | 70 absl::Span<const tensorflow::protobuf_int64> spatial_dims, in GetShapeInfo()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | common_shape_fns.cc | 524 gtl::MutableArraySlice<DimensionHandle> spatial_dims, in DimensionsFromShape()
550 gtl::ArraySlice<DimensionHandle> spatial_dims, in ShapeFromDimensions()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | dynamic_padder.cc | 242 const auto& spatial_dims = in ShouldSkipPadOnOperand() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| H A D | legalize_tf.cc | 984 NamedAttribute GetConvDimensionNumbersAttr(ArrayRef<int64_t> spatial_dims, in GetConvDimensionNumbersAttr()
4944 SmallVector<int64_t, num_spatial_dims> spatial_dims; in matchAndRewrite() local
5168 SmallVector<int64_t, num_spatial_dims> spatial_dims; in matchAndRewrite() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| H A D | legalize_hlo.cc | 754 auto spatial_dims = dnums.getInputSpatialDimensions(); in matchAndRewrite() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| H A D | cuda_dnn.cc | 704 absl::Span<const int64_t> spatial_dims = in CudnnFilterDescriptor() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/ |
| H A D | randomized_tests.cc | 1298 const std::vector<int64_t>& spatial_dims) { in ImageDims()
|