Searched defs:set_output_dim (Results 1 – 2 of 2) sorted by relevance
46 void Splice::set_output_dim(int64_t output_dim) { (void)this->AddAttr(kSpliceOutputDims, api::MakeV… in set_output_dim() function in mindspore::ops::Splice
42 void Affine::set_output_dim(int64_t output_dim) { (void)this->AddAttr(kAffineOutputDim, api::MakeVa… in set_output_dim() function in mindspore::ops::Affine