Searched refs:set_shrink_axis_mask (Results 1 – 6 of 6) sorted by relevance
32 this->set_shrink_axis_mask(shrink_axis_mask); in Init()77 void StridedSliceGrad::set_shrink_axis_mask(int64_t shrink_axis_mask) { in set_shrink_axis_mask() function in mindspore::ops::StridedSliceGrad
42 void set_shrink_axis_mask(int64_t shrink_axis_mask);
43 prim->set_shrink_axis_mask(tflite_attr->shrink_axis_mask); in Parse()
52 void set_shrink_axis_mask(const int64_t shrink_axis_mask);
269 void StridedSlice::set_shrink_axis_mask(const int64_t shrink_axis_mask) { in set_shrink_axis_mask() function in mindspore::ops::StridedSlice283 this->set_shrink_axis_mask(shrink_axis_mask); in Init()
60 prim->set_shrink_axis_mask(attr_value.i()); in Parse()