Home
last modified time | relevance | path

Searched defs:eager_tensor (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/lazy/ts_backend/
H A Dts_native_functions.cpp196 auto eager_tensor = lazy_self->ToTensor(/*detached=*/true); in _to_copy() local
216 auto eager_tensor = lazy_self->ToTensor(/*detached=*/true); in _to_copy() local
H A Dts_eager_fallback.cpp30 at::Tensor eager_tensor = t.to( in _to_eager() local
/aosp_15_r20/external/pytorch/aten/src/ATen/test/
H A Dstride_properties_test.cpp59 auto eager_tensor = tensor.relu(); in TEST() local
/aosp_15_r20/external/tensorflow/tensorflow/python/lib/core/
H A Dpy_func.cc150 tensorflow::Status ExtractTensorFromEagerTensor(const PyObject* eager_tensor, in ExtractTensorFromEagerTensor()
/aosp_15_r20/external/tensorflow/tensorflow/python/framework/experimental/
H A Dunified_api.cc254 TFE_TensorHandle* eager_tensor = EagerTensor_Handle(handle.ptr()); in PYBIND11_MODULE() local
/aosp_15_r20/external/pytorch/torch/csrc/lazy/core/
H A Dtensor.cpp410 auto eager_tensor = in to_lazy_tensor() local