Home
last modified time | relevance | path

Searched refs:spatial_size (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/
Ddnn.cc283 std::copy(spatial_size().begin(), spatial_size().end(), in full_dims()
317 absl::StrAppendFormat(&spatial, "%d ", spatial_size()[i]); in ToString()
335 absl::StrAppendFormat(&spatial, "%d ", spatial_size()[i]); in ToShortString()
366 ret *= spatial_size()[i]; in NodesPerFeatureMap()
Ddnn.h252 int64 height() const { return GetDim(spatial_size(), DimIndex::Y); } in height()
253 int64 width() const { return GetDim(spatial_size(), DimIndex::X); } in width()
254 int64 spatial_dim(DimIndex dim) const { return GetDim(spatial_size(), dim); } in spatial_dim()
255 int ndims() const { return spatial_size().size(); } in ndims()
280 SetDim(spatial_size(), DimIndex::Y, value); in set_height()
284 SetDim(spatial_size(), DimIndex::X, value); in set_width()
288 SetDim(spatial_size(), dim, value); in set_spatial_dim()
339 absl::Span<const int64> spatial_size() const { in spatial_size() function
343 absl::Span<int64> spatial_size() { in spatial_size() function
/external/tensorflow/tensorflow/compiler/xla/service/
Dspace_to_batch_converter.cc72 inherent_high_padding, stride, spatial_size, base_dilation_factor, member
332 VLOG(1) << "spatial size " << c.spatial_size; in IsConvSuitableForSpaceToBatch()
336 if (c.halo_size > CeilOfRatio(c.spatial_size, kNumSplits)) { in IsConvSuitableForSpaceToBatch()
1906 VLOG(1) << "spatial size " << c.spatial_size; in PropagateOnConv()
1922 while (spatial_split_size * num_splits + c.halo_size - c.spatial_size < 0) { in PropagateOnConv()
2625 const int64 spatial_size = in GetConvolutionDetails() local
2643 spatial_size, in GetConvolutionDetails()
2670 VLOG(1) << "spatial size " << c.spatial_size; in PerformSpaceToBatchOnConvolution()
2685 while (spatial_split_size * kNumSplits - c.spatial_size < 0) { in PerformSpaceToBatchOnConvolution()
2709 const int64 pad_size = spatial_split_size * kNumSplits - c.spatial_size; in PerformSpaceToBatchOnConvolution()