- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for out_tensor (0.11 sec)
-
tensorflow/compiler/jit/xla_launch_util_gpu_test.cc
Tensor* CreateHostTensor(const TensorShape& shape, const gtl::ArraySlice<T> data) { Tensor* host_tensor = new Tensor(host_allocator_, DataTypeToEnum<T>::v(), shape); test::FillValues<T>(host_tensor, data); tensors_.push_back(host_tensor); return host_tensor; } // Creates a Tensor on device using the device_allocator_ template <typename T>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 10K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_launch_util_test.cc
Tensor* CreateHostTensor(const TensorShape& shape, const gtl::ArraySlice<T> data) { Tensor* host_tensor = new Tensor(host_allocator_, DataTypeToEnum<T>::v(), shape); test::FillValues<T>(host_tensor, data); tensors_.push_back(host_tensor); return host_tensor; } // Creates a Tensor on device using the device_allocator_ template <typename T>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 09:53:30 UTC 2024 - 28.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/remat/rematerializer.cc
for (const int old_tensor : std::vector<int>(operations_[iop].tensors)) { const auto new_tensor = std::lower_bound(new_tensors.begin(), new_tensors.end(), std::make_pair(old_tensor, 0)); if (new_tensor != new_tensors.end() && new_tensor->first == old_tensor) { DelUse(iop, old_tensor); AddUse(iop, new_tensor->second); } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 14 20:57:44 UTC 2023 - 13.7K bytes - Viewed (0) -
tensorflow/compiler/jit/kernels/xla_ops.cc
args.device_context = new XlaHostRecvDeviceContext( stream, device_memory_base, shape, done_event); Tensor host_tensor; TF_RETURN_IF_ERROR( ctx->rendezvous()->Send(parsed_key, args, host_tensor, false)); return std::move(done_event); }; } // Provide RecvDeviceMemoryFunction for XLA host callbacks. This callback
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 22:46:36 UTC 2024 - 41.4K bytes - Viewed (0)