Searched refs:arg_spec (Results 1 – 6 of 6) sorted by relevance
138 struct arg_spec struct149 typedef arg_spec<157 typedef arg_spec<571 typedef arg_spec<
500 [&enable_sparse](const AbstractBasePtr &arg_spec) -> AbstractBasePtr { in Run() argument501 MS_EXCEPTION_IF_NULL(arg_spec); in Run()502 if (enable_sparse && arg_spec->isa<AbstractTensor>()) { in Run()505 return SensitivityTransform(arg_spec); in Run()
82 bool IsIntermediateAbstract(const AbstractBasePtr &arg_spec) { in IsIntermediateAbstract() argument83 MS_EXCEPTION_IF_NULL(arg_spec); in IsIntermediateAbstract()84 if (dyn_cast<AbstractScalar>(arg_spec)) { in IsIntermediateAbstract()85 auto v = arg_spec->GetValueTrack(); in IsIntermediateAbstract()
148 for (const auto &arg_spec : args_spec_list) { in AbstractJoin() local149 MS_EXCEPTION_IF_NULL(arg_spec); in AbstractJoin()150 arg_spec_tmp = arg_spec_tmp->Join(arg_spec); in AbstractJoin()
160 AbstractTensorPtr arg_spec = CheckArg<AbstractTensor>(op_name, args_spec_list, i); in InferImplBatchNorm() local161 MS_EXCEPTION_IF_NULL(arg_spec); in InferImplBatchNorm()162 MS_EXCEPTION_IF_NULL(arg_spec->shape()); in InferImplBatchNorm()163 ShapeVector arg_shape = arg_spec->shape()->shape(); in InferImplBatchNorm()
10704 my $arg_spec = '';10708 $arg_spec = $Texi2HTML::Config::misc_command{$macro}->{'arg'}10711 if ($arg_spec eq 'line')10717 elsif ($arg_spec)