Home
last modified time | relevance | path

Searched defs:scalar_base (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DRangeFactories.cu142 float scalar_base = static_cast<float>(base); // Use float to avoid promotion to double in logspace_cuda_out() local
156 scalar_t scalar_base = static_cast<scalar_t>(base); in logspace_cuda_out() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DRangeFactories.cpp110 scalar_t scalar_base = static_cast<scalar_t>(base); in logspace_out() local
129 double scalar_base = static_cast<double>(base); // will be autopromoted anyway in logspace_out() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Ddeep_conv2d.cc252 const int64_t scalar_base = vectorized_size * kPacketSize; in operator ()() local