Searched refs:scalar_index (Results 1 – 3 of 3) sorted by relevance
318 constexpr size_t scalar_index = 2; in GetUnsortedSegmentOpScalarArg() local319 if (args_spec_list[scalar_index]->isa<AbstractTensor>()) { // num_segments is Tensor in GetUnsortedSegmentOpScalarArg()320 auto num_segments = args_spec_list[scalar_index]->cast<AbstractTensorPtr>(); in GetUnsortedSegmentOpScalarArg()331 } else if (args_spec_list[scalar_index]->isa<AbstractScalar>()) { // num_segments is Scalar in GetUnsortedSegmentOpScalarArg()332 auto num_segments = CheckArg<AbstractScalar>(op_name, args_spec_list, scalar_index); in GetUnsortedSegmentOpScalarArg()
119 …ValuePtr scalar_index = MakeValue(static_cast<int64_t>(scalar_value->cast<BoolImmPtr>()->value())); in InferImplMakeSlice() local120 slice_args.push_back(scalar_index->ToAbstract()); in InferImplMakeSlice()
597 constexpr size_t scalar_index = 2; in ExpandTuplesC() local598 c_node->input(scalar_index)->set_abstract(aptr); in ExpandTuplesC()