Searched defs:shape_of (Results 1 – 4 of 4) sorted by relevance
/aosp_15_r20/external/pytorch/torch/_inductor/fx_passes/ |
H A D | post_grad.py | 480 def shape_of(bias, a, b): function 488 def cat_tuned_op(match, inputs, dim, *, op, shape_of): argument
|
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/ |
H A D | tensorexpr_fuser.cpp | 432 std::unordered_map<Value*, Value*> shape_of; in buildShapeExpressions() local 527 auto shape_of = buildShapeExpressions(fusion_group); in removeOutputsUsedOnlyInSize() local
|
H A D | graph_fuser.cpp | 931 std::unordered_map<Value*, Value*> shape_of; in buildShapeExpressions() local 1012 auto shape_of = buildShapeExpressions(fusion_group); in removeOutputsUsedOnlyInSize() local
|
/aosp_15_r20/external/libpng/contrib/tools/ |
H A D | genpng.c | 655 shape_of(const char *arg, double width, int f) in shape_of() function
|