Searched defs:spatial_dims (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/ |
| D | hlo_ops_base_structs.cc | 55 ArrayRef<std::pair<IntegerAttr, NonSpatialDim>> non_spatial_dims) { in printConvolutionDimensions()
114 llvm::SmallVector<int64_t> spatial_dims; in parseConvolutionDimensions() local
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| 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
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_grad_shape_utils.h | 47 gtl::InlinedVector<ConvBackpropSpatialDimension, 3> spatial_dims; member
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/experimental/conv_emitter/ |
| D | conv_emitter.cc | 69 absl::Span<const tensorflow::protobuf_int64> spatial_dims, in GetShapeInfo()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_ops.py | 541 spatial_dims=None, argument
719 spatial_dims=None, argument
882 def _with_space_to_batch_adjust(orig, fill_value, spatial_dims): argument
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 524 gtl::MutableArraySlice<DimensionHandle> spatial_dims, in DimensionsFromShape()
550 gtl::ArraySlice<DimensionHandle> spatial_dims, in ShapeFromDimensions()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 1084 DenseIntElementsAttr spatial_dims = in GetConvDimensionNumbersAttr() local
5070 SmallVector<int64_t, num_spatial_dims> spatial_dims; in matchAndRewrite() local
5265 SmallVector<int64_t, num_spatial_dims> spatial_dims; in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 677 const std::vector<int64>& spatial_dims) { in ImageDims()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 641 absl::Span<const int64> spatial_dims = in CudnnFilterDescriptor() local
|