Searched refs:canonical_axis (Results 1 – 2 of 2) sorted by relevance
253 Tidx canonical_axis = axis < 0 ? input_dims + axis : axis; in Compute() local254 OP_REQUIRES(context, canonical_axis >= 0 && canonical_axis < input_dims, in Compute()258 OP_REQUIRES(context, !axes_dense[canonical_axis], in Compute()259 errors::InvalidArgument("axis ", canonical_axis, in Compute()261 axes_dense[canonical_axis] = true; in Compute()
947 int64 canonical_axis = in __anon7c94107b1402() local949 if (canonical_axis < 0 || canonical_axis >= rank) { in __anon7c94107b1402()954 if (axes_dense[canonical_axis]) { in __anon7c94107b1402()955 return errors::InvalidArgument("axis ", canonical_axis, in __anon7c94107b1402()958 axes_dense[canonical_axis] = true; in __anon7c94107b1402()