Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/cuda/
H A DcuSPARSELtOps.cpp275 const auto alpha_tensor = alpha_opt.has_value() ? *alpha_opt: Tensor{}; in _cslt_sparse_mm_impl() local
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dremapper.cc1366 Tensor alpha_tensor; in FindMulAndMaximum() local
2700 Tensor alpha_tensor; in ReplaceMulMaximumWithLeakyRelu() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/nnapi/
H A Dnnapi_delegate.cc4258 TfLiteTensor alpha_tensor; in Map() local