Searched defs:spatial_dims (Results 1 – 9 of 9) sorted by relevance
| /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 | 68 absl::Span<const tensorflow::protobuf_int64> spatial_dims, in GetShapeInfo()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_ops.py | 421 spatial_dims=None, argument
599 spatial_dims=None, argument
762 def _with_space_to_batch_adjust(orig, fill_value, spatial_dims): argument
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 521 gtl::MutableArraySlice<DimensionHandle> spatial_dims, in DimensionsFromShape()
545 gtl::ArraySlice<DimensionHandle> spatial_dims, in ShapeFromDimensions()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 1100 DenseIntElementsAttr spatial_dims = in GetConvDimensionNumbersAttr() local
4166 SmallVector<int64_t, num_spatial_dims> spatial_dims; in matchAndRewrite() local
4361 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 | 594 absl::Span<const int64> spatial_dims = in CudnnFilterDescriptor() local
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 725 const auto& spatial_dims = filter_descriptor.input_filter_dims(); in ScopedFilterDescriptor() local
|