Searched refs:feature_map_input_indexs (Results 1 – 4 of 4) sorted by relevance
229 std::vector<size_t> feature_map_input_indexs; in CreateKernelInfo() local240 feature_map_input_indexs.push_back(index - 1); in CreateKernelInfo()244 feature_map_input_indexs.push_back(index - 1); in CreateKernelInfo()250 if (inputs.size() == 1 || !feature_map_input_indexs.empty()) { in CreateKernelInfo()255 AnfAlgo::SetNodeAttr(kIsFeatureMapInputList, MakeValue(feature_map_input_indexs), cnode_); in CreateKernelInfo()
294 std::vector<size_t> feature_map_input_indexs; in CreateTile() local295 (void)feature_map_input_indexs.emplace_back(0); in CreateTile()296 AnfAlgo::SetNodeAttr(kIsFeatureMapInputList, MakeValue(feature_map_input_indexs), tile_node); in CreateTile()
505 std::vector<int> feature_map_input_indexs; in SetKernelInfoForNode() local507 std::vector<size_t> feature_map_input_indexs; in SetKernelInfoForNode() local514 feature_map_input_indexs.push_back(index); in SetKernelInfoForNode()524 AnfAlgo::SetNodeAttr(kIsFeatureMapInputList, MakeValue(feature_map_input_indexs), node); in SetKernelInfoForNode()
705 std::vector<size_t> feature_map_input_indexs; in CreateCNode() local710 feature_map_input_indexs.push_back(i); in CreateCNode()720 SetNodeAttrSafely(kIsFeatureMapInputList, MakeValue(feature_map_input_indexs), cnode); in CreateCNode()