Searched refs:canonical_axis (Results 1 – 2 of 2) sorted by relevance
250 Tidx canonical_axis = axis < 0 ? input_dims + axis : axis; in Compute() local251 OP_REQUIRES(context, canonical_axis >= 0 && canonical_axis < input_dims, in Compute()255 OP_REQUIRES(context, !axes_dense[canonical_axis], in Compute()256 errors::InvalidArgument("axis ", canonical_axis, in Compute()258 axes_dense[canonical_axis] = true; in Compute()
1014 int64 canonical_axis = in __anon42d741191402() local1016 if (canonical_axis < 0 || canonical_axis >= rank) { in __anon42d741191402()1021 if (axes_dense[canonical_axis]) { in __anon42d741191402()1022 return errors::InvalidArgument("axis ", canonical_axis, in __anon42d741191402()1025 axes_dense[canonical_axis] = true; in __anon42d741191402()