Searched defs:dim_val (Results 1 – 14 of 14) sorted by relevance
130 const auto dim_val = dim.value(); in get_out_numel() local175 const auto dim_val = dim.value(); in get_init_index() local231 const auto dim_val = dim.value(); in compute_reduced_out_size() local
189 auto dim_val = *intValue(multidimbuf->dim(d)); in PrintArguments() local
2279 auto dim_val = this->value_; in visit() local2450 auto dim_val = this->value_; in visit() local
81 int64_t dim_val = dim.to<int64_t>(); in __anone4f01e490202() local
143 int dim_val = 0; in toList() local
267 int64_t dim_val = Value(Dim(s, i)); in NumElements() local
2223 auto dim_val = c->Value(dim); in SliceHelper() local
943 auto dim_val = graph->insertConstant(dim); in getSizeOf() local
1932 Value* dim_val = insertConstant(IValue(dim)); in insertToList() local
2506 int64_t dim_val = (*dim_attr.begin()).getSExtValue(); in InferExpandDimsOpType() local
1602 int64_t dim_val = dim.value().getSExtValue(); in verify() local
6355 def prim_tolist(g: jit_utils.GraphContext, input, dim_val, elem_ty_val): argument
3517 int64_t dim_val = dim.value().getSExtValue(); in matchAndRewrite() local
8725 def __init__(self, ord_val, dim_val=(-2, -1), keepdim_val=False): argument