Searched refs:input_ctrl_tensors (Results 1 – 3 of 3) sorted by relevance
139 const auto &input_ctrl_tensors = kernel_graph_->input_ctrl_tensors(); in SetOpMappingInfo() local141 …bool valid_ctrl_tensors = input_ctrl_tensors != nullptr && input_ctrl_tensors->size() >= kLoopSink… in SetOpMappingInfo()180 const auto ¤t_step_tensor = input_ctrl_tensors->at(kCurrentStepTensorIndex); in SetOpMappingInfo()181 const auto &currnet_epoch_tensor = input_ctrl_tensors->at(kCurrentEpochTensorIndex); in SetOpMappingInfo()182 const auto &steps_per_epoch_tensor = input_ctrl_tensors->at(kStepsPerEpochTensorIndex); in SetOpMappingInfo()
209 …std::shared_ptr<std::vector<tensor::TensorPtr>> input_ctrl_tensors() const { return input_ctrl_ten… in input_ctrl_tensors() function
225 auto inputs_params = graph->input_ctrl_tensors(); in LoadCtrlInputTensor()293 if (graph->input_ctrl_tensors()) { in UpdateCtrlInputTensor()