Home
last modified time | relevance | path

Searched refs:CustomDeviceTensorHandle (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/eager/
Dcustom_device_op_handler.cc77 if (tensorflow::CustomDeviceTensorHandle::classof(inputs[i])) { in Execute()
78 tensorflow::CustomDeviceTensorHandle* previous = in Execute()
79 tensorflow::down_cast<tensorflow::CustomDeviceTensorHandle*>( in Execute()
118 if (CustomDeviceTensorHandle::classof(generic_input)) { in MaybePinToCustomDevice()
119 const CustomDeviceTensorHandle* input = in MaybePinToCustomDevice()
120 down_cast<const CustomDeviceTensorHandle*>(generic_input); in MaybePinToCustomDevice()
140 if (CustomDeviceTensorHandle::classof(generic_input)) { in MaybePinToCustomDevice()
141 const CustomDeviceTensorHandle* input = in MaybePinToCustomDevice()
142 down_cast<const CustomDeviceTensorHandle*>(generic_input); in MaybePinToCustomDevice()
Dcustom_device.cc20 Status CustomDeviceTensorHandle::Shape(PartialTensorShape* shape) const { in Shape()
30 Status CustomDeviceTensorHandle::NumElements(int64* num_elements) const { in NumElements()
42 const char* CustomDeviceTensorHandle::DeviceType(Status* status) const { in DeviceType()
50 int CustomDeviceTensorHandle::DeviceId(Status* status) const { in DeviceId()
58 AbstractTensorInterface* CustomDeviceTensorHandle::Resolve(Status* status) { in Resolve()
71 const DeviceNameUtils::ParsedName* CustomDeviceTensorHandle::ParsedName( in ParsedName()
Dcustom_device.h30 class CustomDeviceTensorHandle; variable
72 class CustomDeviceTensorHandle : public ImmediateExecutionTensorHandle {
74 CustomDeviceTensorHandle(ImmediateExecutionContext* context, in CustomDeviceTensorHandle() function
Dcustom_device_test.cc64 class TestCustomDeviceTensorHandle : public CustomDeviceTensorHandle {
69 : CustomDeviceTensorHandle(context, device, dtype) {} in TestCustomDeviceTensorHandle()
Deager_operation.cc271 if (CustomDeviceTensorHandle::classof(h)) { in AddInput()
283 if (CustomDeviceTensorHandle::classof(input)) { in AddInputList()
300 if (CustomDeviceTensorHandle::classof(previous)) { in SetInput()
303 if (CustomDeviceTensorHandle::classof(input)) { in SetInput()
/external/tensorflow/tensorflow/c/eager/
Dc_api.cc372 if (tensorflow::CustomDeviceTensorHandle::classof(unwrapped_handle)) { in TFE_TensorHandleDevicePointer()
373 return tensorflow::down_cast<tensorflow::CustomDeviceTensorHandle*>( in TFE_TensorHandleDevicePointer()
487 : public tensorflow::CustomDeviceTensorHandle {
499 : tensorflow::CustomDeviceTensorHandle(context, device, dtype), in CAPICustomDeviceTensorHandle()
Dc_api_experimental.cc617 if (tensorflow::CustomDeviceTensorHandle::classof(unwrapped_handle)) { in TFE_CreatePackedTensorHandle()
621 tensorflow::down_cast<tensorflow::CustomDeviceTensorHandle*>( in TFE_CreatePackedTensorHandle()