Home
last modified time | relevance | path

Searched defs:op_device (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/eager/
Dtensor_handle.cc48 TensorHandle::TensorHandle(const class Tensor& t, Device* d, Device* op_device, in TensorHandle()
62 TensorHandle::TensorHandle(uint64 node_id, Device* d, Device* op_device, in TensorHandle()
84 Device* op_device, Device* resource_device, in TensorHandle()
160 tensorflow::Device** op_device) { in TensorAndDevice()
Dtensor_handle.h85 tensorflow::Device* op_device() const { return op_device_; } in op_device() function
Dexecute.cc294 bool IsMultiDevice(const FunctionDef* fdef, const string& op_device) { in IsMultiDevice()
823 Device* op_device = op->Device() == nullptr ? ctx->HostCPU() : op->Device(); in MaybeUpdateOpDevice() local
/external/tensorflow/tensorflow/c/
Dc_test_util.cc210 TF_Operation** op, const string& op_device, bool check) { in BinaryOpHelper()
225 const string& op_device, TF_Status* s, in MinWithDevice()
/external/tensorflow/tensorflow/c/eager/
Dc_api.cc524 tensorflow::Device* op_device = nullptr; in TFE_TensorHandleResolve() local
884 tensorflow::Device* op_device = nullptr; in TFE_TensorHandleUnderlyingTensorInHostMemory() local