Searched refs:current_step_tensor (Results 1 – 1 of 1) sorted by relevance
180 const auto ¤t_step_tensor = input_ctrl_tensors->at(kCurrentStepTensorIndex); in SetOpMappingInfo() local184 MS_EXCEPTION_IF_NULL(current_step_tensor); in SetOpMappingInfo()187 MS_EXCEPTION_IF_NULL(current_step_tensor->device_address()); in SetOpMappingInfo()191 void *current_step = current_step_tensor->device_address()->GetMutablePtr(); in SetOpMappingInfo()