Searched defs:TensorDescriptor (Results 1 – 3 of 3) sorted by relevance
45 struct TensorDescriptor : public GPUObjectDescriptor { struct47 TensorDescriptor(DataType dt, TensorStorageType st, Layout l) in TensorDescriptor() argument51 TensorDescriptor& operator=(const TensorDescriptor&) = default; argument53 TensorDescriptor& operator=(TensorDescriptor&& desc); argument70 void Release() override { data.clear(); } in Release()95 DataType data_type = DataType::UNKNOWN;96 TensorStorageType storage_type = TensorStorageType::UNKNOWN;99 Layout layout =103 BHWDC shape;104 std::vector<uint8_t> data;
76 TensorDescriptor::TensorDescriptor(TensorDescriptor&& desc) in TensorDescriptor() function in tflite::gpu::TensorDescriptor
428 using TensorDescriptor = typedef