Searched refs:index_dim (Results 1 – 6 of 6) sorted by relevance
492 assert(dst->base.index_dim == 1); in dcl_vs_input()506 if (dst->base.index_dim == 2) { in dcl_gs_input()524 assert(dst->base.index_dim == 0); in dcl_gs_input()537 assert(dst->base.index_dim == 1); in dcl_sgv_input()553 assert(dst->base.index_dim == 2); in dcl_siv_input()573 assert(dst->base.index_dim == 1); in dcl_ps_input()592 assert(dst->base.index_dim == 1); in dcl_ps_sgv_input()633 assert(dst->base.index_dim == 1); in dcl_ps_siv_input()715 assert(operand->index_dim == 1); in translate_operand()723 assert(operand->index_dim == 1); in translate_operand()[all …]
197 operand->index_dim = 0; in parse_operand()200 operand->index_dim = 1; in parse_operand()203 operand->index_dim = 2; in parse_operand()209 if (operand->index_dim >= 1) { in parse_operand()211 if (operand->index_dim >= 2) { in parse_operand()276 if (operand->index_dim >= 1) { in parse_operand_index()278 if (operand->index_dim >= 2) { in parse_operand_index()
93 unsigned index_dim; member
67 def __init__(self, index_dim, strategy, index_size=16): argument72 if index_dim == 1:76 if index_dim == 2:199 def __init__(self, index_dim, strategy, index_size=16): argument204 if index_dim == 1:208 if index_dim == 2:
110 index_dim = static_cast<int>(input0->shape().size()); in GenerateIndices()113 switch (index_dim) { in GenerateIndices()148 MS_LOG(ERROR) << "Indices dimensions is " << index_dim << ", which must be 0, 1 or 2"; in GenerateIndices()
57 int index_dim = 0; variable