Searched refs:LocalTensorHandleData (Results 1 – 4 of 4) sorted by relevance
24 Status LocalTensorHandleData::Tensor(const tensorflow::Tensor** t) const { in Tensor()30 Status LocalTensorHandleData::TensorValue(tensorflow::TensorValue* t) { in TensorValue()37 Status LocalTensorHandleData::Shape(TensorShape* shape) const { in Shape()43 Status LocalTensorHandleData::NumDims(int* num_dims) const { in NumDims()49 Status LocalTensorHandleData::Dim(int dim_index, int64* dim) const { in Dim()55 Status LocalTensorHandleData::NumElements(int64* num_elements) const { in NumElements()
42 class LocalTensorHandleData : public TensorHandleData {44 explicit LocalTensorHandleData(const tensorflow::Tensor& t) : tensor_(t) {} in LocalTensorHandleData() function45 ~LocalTensorHandleData() override {} in ~LocalTensorHandleData()
111 *h = new TensorHandle(absl::make_unique<LocalTensorHandleData>(t), in CreateLocalHandle()115 *h = new TensorHandle(absl::make_unique<LocalTensorHandleData>(t), in CreateLocalHandle()122 TensorHandle::TensorHandle(std::unique_ptr<LocalTensorHandleData> t, in TensorHandle()141 TensorHandle::TensorHandle(std::unique_ptr<LocalTensorHandleData> t, in TensorHandle()572 tensor_handle_data_ = absl::make_unique<LocalTensorHandleData>(tensor); in SetTensor()
64 TensorHandle(std::unique_ptr<LocalTensorHandleData> t, DataType dtype,67 TensorHandle(std::unique_ptr<LocalTensorHandleData> t,