Searched refs:_axis (Results 1 – 4 of 4) sorted by relevance
53 const Tensor* _axis = c->input_tensor(2); in ReduceSliceShapeFn() local54 if (nullptr == _axis) { in ReduceSliceShapeFn()57 int64 axis = _axis->scalar<int64>()(); in ReduceSliceShapeFn()
134 const Tensor& _axis = context->input(2); in Compute() local135 int64 axis = _axis.scalar<int64>()(); in Compute()
9234 auto _axis = _o->axis;9238 _axis,9838 auto _axis = _o->axis;9841 _axis);10645 auto _axis = _o->axis;10649 _axis);10695 auto _axis = _o->axis;10698 _axis);10792 auto _axis = _o->axis;10796 _axis);
264 .. _axis-descriptor-object: