Home
last modified time | relevance | path

Searched refs:CustomDeviceTensorHandle (Results 1 – 8 of 8) 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()
151 if (CustomDeviceTensorHandle::classof(generic_input)) { in MaybePinToCustomDevice()
152 const CustomDeviceTensorHandle* input = in MaybePinToCustomDevice()
153 down_cast<const CustomDeviceTensorHandle*>(generic_input); in MaybePinToCustomDevice()
173 if (CustomDeviceTensorHandle::classof(generic_input)) { in MaybePinToCustomDevice()
174 const CustomDeviceTensorHandle* input = in MaybePinToCustomDevice()
175 down_cast<const CustomDeviceTensorHandle*>(generic_input); in MaybePinToCustomDevice()
Dcustom_device.cc22 Status CustomDeviceTensorHandle::Shape(PartialTensorShape* shape) const { in Shape()
32 Status CustomDeviceTensorHandle::NumElements(int64_t* num_elements) const { in NumElements()
50 const char* CustomDeviceTensorHandle::DeviceType(Status* status) const { in DeviceType()
58 int CustomDeviceTensorHandle::DeviceId(Status* status) const { in DeviceId()
66 AbstractTensorInterface* CustomDeviceTensorHandle::Resolve(Status* status) { in Resolve()
79 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
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()
Dcustom_device_test.cc67 class TestCustomDeviceTensorHandle : public CustomDeviceTensorHandle {
72 : CustomDeviceTensorHandle(context, device, dtype), length_(length) {} in TestCustomDeviceTensorHandle()
/external/tensorflow/tensorflow/c/eager/
Dc_api.cc375 if (tensorflow::CustomDeviceTensorHandle::classof(unwrapped_handle)) { in TFE_TensorHandleDevicePointer()
376 return tensorflow::down_cast<tensorflow::CustomDeviceTensorHandle*>( in TFE_TensorHandleDevicePointer()
490 : public tensorflow::CustomDeviceTensorHandle {
496 : tensorflow::CustomDeviceTensorHandle(context, device, dtype), in CAPICustomDeviceTensorHandle()
519 return tensorflow::CustomDeviceTensorHandle::SummarizeValue(summary); in SummarizeValue()
Dc_api_experimental.cc673 if (tensorflow::CustomDeviceTensorHandle::classof(unwrapped_handle)) { in TFE_CreatePackedTensorHandle()
677 tensorflow::down_cast<tensorflow::CustomDeviceTensorHandle*>( in TFE_CreatePackedTensorHandle()
/external/tensorflow/tensorflow/core/tfrt/eager/
Dc_api_tfrt.cc1507 if (tensorflow::CustomDeviceTensorHandle::classof(h)) { in AddInput()
1524 if (tensorflow::CustomDeviceTensorHandle::classof(args_[index].get())) { in SetInput()
1527 if (tensorflow::CustomDeviceTensorHandle::classof(input)) { in SetInput()