Searched defs:node_device_tensor (Results 1 – 1 of 1) sorted by relevance
32 inline void UpdateShape(const AnfNodePtr &input_node, const DeviceTensorPtr &node_device_tensor, in UpdateShape()49 … const DeviceTensorPtr &node_device_tensor, const KernelTransformType &type) { in InputDataNoNeedCopy()627 … const DeviceTensorPtr &node_device_tensor, size_t i) { in CopyInputDataPersistedHandle()704 auto &node_device_tensor = node_device_tensors_[i]; in CopyInputData() local