Searched defs:expand_dims (Results 1 – 7 of 7) sorted by relevance
64 auto expand_dims = func_graph->NewCNode(expand_dims_inputs); in AddBroadCastToNode() local
50 auto expand_dims = VectorRef({is_expand_dims, mask_input, is_param1}); in DefineMask() local
782 auto expand_dims = gen_g.PushBack({gen_g.NewOpInst(EXPAND_DIMS), cast, CreatInt64Imm(axis_ - 1)}); in ComputeReplaceGraph() local
131 expand_dims = P.ExpandDims() variable
574 def expand_dims(x, axis): function
40 def expand_dims(a, axis): function
544 ExpandDims expand_dims{}; in ConvertExpandDims() local