Method SetValue

src/core/tensor/tensor.cc:849–862

template <typename SType>
void Tensor::SetValue(const SType x) {
  // auto size = Size();
  auto ptr = block_;

  TYPE_LANG_SWITCH(data_type_, DType, device_->lang(), Lang, {
    DType tmp = TypeCast<SType, DType>(x);
    Tensor &thisRef = *this;
    device_->Exec(
        [thisRef, tmp](Context *ctx) mutable {
          Set<DType, Lang>(tmp, &thisRef, ctx);
        },
        {}, {ptr}, "SetValue");
  });
}
template void Tensor::SetValue<float>(const float x);
template void Tensor::SetValue<half_float::half>(const half_float::half x);
template void Tensor::SetValue<int>(const int x);