Home
last modified time | relevance | path

Searched defs:value_scalar (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DFillKernel.cpp17 void fill_non_native_type(TensorIterator& iter, const Scalar& value_scalar) { in fill_non_native_type()
29 void fill_non_native_type<c10::complex<at::Half>>(TensorIterator& iter, const Scalar& value_scalar)… in fill_non_native_type()
39 void fill_kernel(TensorIterator& iter, const Scalar& value_scalar) { in fill_kernel()
H A DActivation.cpp150 const Scalar& value_scalar) { in threshold_kernel()
/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DScalarOps.cpp10 inline void fill_inplace(Tensor& self, const Scalar& value_scalar) { in fill_inplace()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dreader_ops.cc94 auto value_scalar = value->scalar<tstring>(); in ComputeWithReader() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/kernels/
H A DQuantizedOpKernels.cpp1028 const Scalar& value_scalar, in qthreshold_kernel()