Searched defs:_axis (Results 1 – 2 of 2) sorted by relevance
53 const Tensor* _axis = c->input_tensor(2); in ReduceSliceShapeFn() local
134 const Tensor& _axis = context->input(2); in Compute() local