Searched defs:orig_shape (Results 1 – 9 of 9) sorted by relevance
87 mxnet::TShape orig_shape = in_data[l2_normalization::kData].shape_; in Forward() local158 mxnet::TShape orig_shape = out_data[l2_normalization::kOut].shape_; in Backward() local
50 mxnet::TShape orig_shape = in_data[l2_normalization::kData].shape_; in Forward() local
293 Expr orig_shape = src_shape[i]; in TransformShape() local
292 PrimExpr orig_shape = src_shape[i]; in TransformShape() local
1897 let orig_shape = shape.offset_left(last_line_width + 1).unwrap_or(Shape { in rewrite_assign_rhs_expr() localVariable
4366 def _insert_axis_in_shape(context, builder, orig_shape, ndim, axis): argument