Searched refs:set_new_axis_mask (Results 1 – 6 of 6) sorted by relevance
31 this->set_new_axis_mask(new_axis_mask); in Init()68 void StridedSliceGrad::set_new_axis_mask(int64_t new_axis_mask) { in set_new_axis_mask() function in mindspore::ops::StridedSliceGrad
41 void set_new_axis_mask(int64_t new_axis_mask);
42 prim->set_new_axis_mask(tflite_attr->new_axis_mask); in Parse()
50 void set_new_axis_mask(const int64_t new_axis_mask);
261 void StridedSlice::set_new_axis_mask(const int64_t new_axis_mask) { in set_new_axis_mask() function in mindspore::ops::StridedSlice282 this->set_new_axis_mask(new_axis_mask); in Init()
54 prim->set_new_axis_mask(attr_value.i()); in Parse()