Searched refs:LocalTensorHandleData (Results 1 – 4 of 4) sorted by relevance
25 Status LocalTensorHandleData::Tensor(const tensorflow::Tensor** t) const { in Tensor()33 Status LocalTensorHandleData::TensorValue(tensorflow::TensorValue* t) { in TensorValue()42 Status LocalTensorHandleData::Shape(TensorShape* shape) const { in Shape()50 Status LocalTensorHandleData::NumDims(int* num_dims) const { in NumDims()58 Status LocalTensorHandleData::Dim(int dim_index, int64_t* dim) const { in Dim()66 Status LocalTensorHandleData::NumElements(int64_t* num_elements) const { in NumElements()74 Status LocalTensorHandleData::Unprotect() { in Unprotect()84 Status LocalTensorHandleData::SetTensor(tensorflow::Tensor&& t) { in SetTensor()97 string LocalTensorHandleData::DebugString() const { in DebugString()105 void LocalTensorHandleData::BlockingControl::SetReady() { in SetReady()[all …]
26 class LocalTensorHandleData {28 LocalTensorHandleData() : ctrl_(absl::in_place_type<BlockingControl>) {} in LocalTensorHandleData() function29 explicit LocalTensorHandleData(tensorflow::Tensor&& t) in LocalTensorHandleData() function
192 auto& data = absl::get<LocalTensorHandleData>(data_); in GetResourceHandleDtypesAndShapes()252 data_(absl::in_place_type<LocalTensorHandleData>, std::move(t)) { in TensorHandle()271 data_(absl::in_place_type<LocalTensorHandleData>, std::move(t)) { in TensorHandle()296 data_(absl::in_place_type<LocalTensorHandleData>) { in TensorHandle()465 auto& data = absl::get<LocalTensorHandleData>(data_); in Tensor()479 auto& data = absl::get<LocalTensorHandleData>(data_); in TensorFromDevice()503 auto& data = absl::get<LocalTensorHandleData>(data_); in TensorValue()937 auto& data = absl::get<LocalTensorHandleData>(data_); in SetTensor()
313 std::unordered_map<const tensorflow::Device*, LocalTensorHandleData>385 absl::variant<LocalTensorHandleData, PackedTensorHandleData,389 absl::variant<LocalTensorHandleData, PackedTensorHandleData> data_;