Searched refs:steps_per_epoch_tensor (Results 1 – 1 of 1) sorted by relevance
182 const auto &steps_per_epoch_tensor = input_ctrl_tensors->at(kStepsPerEpochTensorIndex); in SetOpMappingInfo() local186 MS_EXCEPTION_IF_NULL(steps_per_epoch_tensor); in SetOpMappingInfo()189 MS_EXCEPTION_IF_NULL(steps_per_epoch_tensor->device_address()); in SetOpMappingInfo()193 void *steps_per_epoch = steps_per_epoch_tensor->device_address()->GetMutablePtr(); in SetOpMappingInfo()