Searched refs:axes_ (Results 1 – 2 of 2) sorted by relevance
53 axes_ = std::vector<int32_t>(axes.begin(), axes.end()); in Axes()57 inline const std::vector<int32_t>& Axes() const { return axes_; } in Axes()108 std::vector<int32_t> axes_; variable
890 for axes_ in input_axis_labels:891 for ax in axes_:1098 for i, (input_, axes_) in enumerate(zip(inputs, idx_in)):1099 if input_.shape.ndims != len(axes_):1103 i, axes_, len(axes_), input_.shape.ndims1107 sorted_idx = sorted(axes_, key=axis_order.get)1109 if len(set(axes_)) != len(axes_):1111 'Subscript not supported: an axis appears more than once: %s' % axes_)1113 if list(axes_) != sorted_idx:1114 permuted = [axes_.find(ax) for ax in sorted_idx]