Home
last modified time | relevance | path

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

/aosp_15_r20/external/ComputeLibrary/examples/
H A Dgraph_mobilenet.cpp72 …unsigned int spatial_size = (model_id == 0 || common_params.data_type == DataType::QASYMM8) ? 224 … in do_setup() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DGridSamplerKernel.cpp1150 auto spatial_size = H * W; in grid_sampler_2d_cpu_kernel_impl() local
1233 auto spatial_size = grid.size(1) * grid.size(2); in grid_sampler_2d_backward_cpu_kernel_impl() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/
H A Ddnn.h359 absl::Span<const int64_t> spatial_size() const { in spatial_size() function
363 absl::Span<int64_t> spatial_size() { in spatial_size() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dspace_to_batch_converter.cc73 int64_t inherent_low_padding, inherent_high_padding, stride, spatial_size, member
3589 const int64_t spatial_size = input_dim_size + in GetConvolutionDetails() local