Home
last modified time | relevance | path

Searched refs:shape_funcs (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/
H A Dserialized_shape_function_registry.cpp19 std::string shape_funcs = "" variable
3253 return shape_funcs; in GetSerializedShapeFunctions()
/aosp_15_r20/external/pytorch/test/torch_np/
H A Dtest_basic.py253 shape_funcs = [w.zeros, w.empty, w.ones, functools.partial(w.full, fill_value=42)] variable
262 @parametrize("func", shape_funcs)
/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dcommon_methods_invocations.py24419 shape_funcs = [op for op in ops_and_refs if isinstance(op, ShapeFuncInfo)] variable