Searched defs:dim_indices (Results 1 – 2 of 2) sorted by relevance
611 def _propagate(dim_indices, conf, cells, c_prev, m_prev, new_output, new_state, argument
251 std::vector<int64> PickDims(gtl::ArraySlice<int64> dim_indices) const { in PickDims()