Home
last modified time | relevance | path

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

/external/ComputeLibrary/examples/
Dgraph_mobilenet.cpp72 …unsigned int spatial_size = (model_id == 0 || common_params.data_type == DataType::QASYMM8) ? 224 … in do_setup() local
75 …const TensorShape tensor_shape = permute_shape(TensorShape(spatial_size, spatial_size, 3U, 1U)… in do_setup()
/external/tensorflow/tensorflow/stream_executor/
Ddnn.cc371 std::copy(spatial_size().begin(), spatial_size().end(), in full_dims()
420 absl::StrAppendFormat(&spatial, "%d ", spatial_size()[i]); in ToString()
438 absl::StrAppendFormat(&spatial, "%d ", spatial_size()[i]); in ToShortString()
469 ret *= spatial_size()[i]; in NodesPerFeatureMap()
Ddnn.h266 int64 height() const { return GetDim(spatial_size(), DimIndex::Y); } in height()
267 int64 width() const { return GetDim(spatial_size(), DimIndex::X); } in width()
268 int64 spatial_dim(DimIndex dim) const { return GetDim(spatial_size(), dim); } in spatial_dim()
269 int ndims() const { return spatial_size().size(); } in ndims()
303 SetDim(spatial_size(), DimIndex::Y, value); in set_height()
307 SetDim(spatial_size(), DimIndex::X, value); in set_width()
311 SetDim(spatial_size(), dim, value); in set_spatial_dim()
362 absl::Span<const int64> spatial_size() const { in spatial_size() function
366 absl::Span<int64> spatial_size() { in spatial_size() function
/external/tensorflow/tensorflow/compiler/xla/service/
Dspace_to_batch_converter.cc74 inherent_high_padding, stride, spatial_size, base_dilation_factor, member
395 VLOG(1) << "spatial size " << c.spatial_size << " halo size " << c.halo_size; in IsConvSuitableForSpaceToBatch()
399 if (c.halo_size > CeilOfRatio(c.spatial_size, ctrl_.number_of_splits)) { in IsConvSuitableForSpaceToBatch()
2276 VLOG(1) << "spatial size " << c.spatial_size << " halo size " << c.halo_size in PropagateOnConv()
2285 while (spatial_split_size * num_splits + c.halo_size - c.spatial_size < 0 || in PropagateOnConv()
3090 const int64_t spatial_size = in GetConvolutionDetails() local
3108 spatial_size, in GetConvolutionDetails()
3140 VLOG(1) << "spatial size " << c.spatial_size; in PerformSpaceToBatchOnConvolution()
3143 if (c.spatial_size < 2 * ctrl_.number_of_splits) { in PerformSpaceToBatchOnConvolution()
3160 while (spatial_split_size * ctrl_.number_of_splits - c.spatial_size < 0) { in PerformSpaceToBatchOnConvolution()
[all …]