Searched defs:spatial_dims (Results 1 – 5 of 5) sorted by relevance
328 spatial_dims=None, argument505 spatial_dims=None, argument657 def _with_space_to_batch_adjust(orig, fill_value, spatial_dims): argument
354 gtl::MutableArraySlice<DimensionHandle> spatial_dims, in DimensionsFromShape()378 gtl::ArraySlice<DimensionHandle> spatial_dims, in ShapeFromDimensions()
676 const std::vector<int64>& spatial_dims) { in ImageDims()
572 absl::Span<const int64> spatial_dims = in CudnnFilterDescriptor() local
614 const auto& spatial_dims = filter_descriptor.input_filter_dims(); in ScopedFilterDescriptor() local