Home
last modified time | relevance | path

Searched defs:params_spec (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_functorch/_aot_autograd/
H A Dtraced_function_transforms.py844 def create_functional_call(mod, params_spec, params_len, store_orig_mod=False): argument
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/expansions/
H A Dgather_spmd_expander.cc321 const ShardingSpec& params_spec = params_layout->dim(i); in GatherNdGetOutputLayoutFromInput() local