Searched refs:TensorHandleInterface (Results 1 – 3 of 3) sorted by relevance
357 TensorHandleInterface::TensorHandleInterface(Value&& v, TfrtContext* context) in TensorHandleInterface() function in tfrt::tf::TensorHandleInterface362 TensorHandleInterface::TensorHandleInterface(tensorflow::DataType dtype, in TensorHandleInterface() function in tfrt::tf::TensorHandleInterface369 tensorflow::DataType TensorHandleInterface::DataType() const { in DataType()402 tensorflow::Status TensorHandleInterface::TensorHandleStatus() const { in TensorHandleStatus()430 tensorflow::Status TensorHandleInterface::Shape( in Shape()447 tensorflow::Status TensorHandleInterface::NumDims(int* num_dims) const { in NumDims()458 tensorflow::Status TensorHandleInterface::NumElements( in NumElements()470 tensorflow::Status TensorHandleInterface::Dim(int dim_index, in Dim()482 const char* TensorHandleInterface::DeviceName( in DeviceName()495 const char* TensorHandleInterface::BackingDeviceName( in BackingDeviceName()[all …]
353 class TensorHandleInterface356 explicit TensorHandleInterface(Value&& v, TfrtContext* context);358 explicit TensorHandleInterface(tensorflow::DataType dtype, Value&& v,425 inline TensorHandleInterface* TensorHandleFromInterface(T* handle) { in TensorHandleFromInterface()426 return tensorflow::down_cast<TensorHandleInterface*>(handle); in TensorHandleFromInterface()
398 class TensorHandleInterface : public ImmediateExecutionTensorHandle {