Searched refs:reserved_space_tensor (Results 1 – 1 of 1) sorted by relevance
745 Tensor* reserved_space_tensor = nullptr; in Compute() local794 &reserved_space_tensor); in Compute()796 wksp.SetUsrMem(workspace_md, reserved_space_tensor); in Compute()806 &reserved_space_tensor); in Compute()957 Tensor* reserved_space_tensor = nullptr; in HandleEmptyInput() local961 &reserved_space_tensor); in HandleEmptyInput()970 Tensor** reserved_space_tensor) { in AllocateTFOutputs() argument1032 DCHECK(reserved_space_tensor != nullptr); in AllocateTFOutputs()1037 reserved_space_tensor, workspace_tf_shape, in AllocateTFOutputs()1039 DCHECK((*reserved_space_tensor) != nullptr); in AllocateTFOutputs()[all …]