Searched defs:packed_tensor (Results 1 – 6 of 6) sorted by relevance
51 …ICLTensor *packed_tensor = utils::cast::polymorphic_downcast<ICLTensor *>(pack.get_tensor(slot_id)… in _tensor() local
51 … ITensor *packed_tensor = utils::cast::polymorphic_downcast<ITensor *>(pack.get_tensor(slot_id)); in _tensor() local
63 at::Tensor unpack_rgb(const at::Tensor& packed_tensor) { in unpack_rgb()90 const at::Tensor& packed_tensor // OUT in pack_rgb()
232 TFE_TensorHandle* packed_tensor; in PYBIND11_MODULE() local
327 PyObject* packed_tensor = in InputTFE_OutputTensorHandles() local
832 std::unique_ptr<TensorWithLayout> packed_tensor; in Pack() local990 std::unique_ptr<TensorWithLayout> packed_tensor; in SparsePack() local