Searched refs:eager_tensor (Results 1 – 8 of 8) sorted by relevance
| /external/pytorch/test/jit/ |
| D | test_generator.py | 43 eager_tensor = f()
51 self.assertEqual(eager_tensor, traced_tensor)
70 eager_tensor = f()
79 self.assertEqual(eager_tensor, script_tensor)
91 eager_tensor = f()
98 self.assertEqual(eager_tensor, script_tensor)
117 eager_tensor = f(generator)
126 self.assertEqual(eager_tensor, script_tensor)
|
| /external/pytorch/aten/src/ATen/test/ |
| D | stride_properties_test.cpp | 59 auto eager_tensor = tensor.relu(); in TEST() local
60 …auto ref_type = TensorType::create(std::nullopt, std::nullopt, eager_tensor.sizes(), eager_tensor.… in TEST()
|
| /external/pytorch/torch/_lazy/ |
| D | extract_compiled_graph.py | 91 for uniq_idx, eager_tensor in enumerate(eager_tensor_list):
93 duplicated_list[dup_idx] = eager_tensor
|
| /external/pytorch/torch/csrc/lazy/ts_backend/ |
| D | ts_native_functions.cpp | 196 auto eager_tensor = lazy_self->ToTensor(/*detached=*/true); in _to_copy() local
199 eager_tensor.to(options, /*non_blocking=*/non_blocking, /*copy=*/true); in _to_copy()
216 auto eager_tensor = lazy_self->ToTensor(/*detached=*/true); in _to_copy() local
224 eager_tensor.to(options, /*non_blocking=*/false, /*copy=*/true); in _to_copy()
|
| D | ts_eager_fallback.cpp | 30 at::Tensor eager_tensor = t.to( in _to_eager() local
34 eager_tensors.push_back(eager_tensor); in _to_eager()
|
| /external/tensorflow/tensorflow/python/framework/experimental/ |
| D | unified_api.cc | 254 TFE_TensorHandle* eager_tensor = EagerTensor_Handle(handle.ptr()); in PYBIND11_MODULE() local
255 auto t = static_cast<AbstractTensorHandle*>(unwrap(eager_tensor)); in PYBIND11_MODULE()
|
| /external/tensorflow/tensorflow/python/lib/core/ |
| D | py_func.cc | 150 tensorflow::Status ExtractTensorFromEagerTensor(const PyObject* eager_tensor, in ExtractTensorFromEagerTensor() argument
156 tensorflow::unwrap(EagerTensor_Handle(eager_tensor)))); in ExtractTensorFromEagerTensor()
|
| /external/pytorch/torch/csrc/lazy/core/ |
| D | tensor.cpp | 410 auto eager_tensor = in to_lazy_tensor() local
413 eager_tensor, torch::lazy::atenDeviceToBackendDevice(device)); in to_lazy_tensor()
|