Searched defs:out_spec (Results 1 – 7 of 7) sorted by relevance
19 def _unlift(gm, inp_pos_to_param_buffer_name, in_spec, out_spec, state_dict): argument
193 const char* out_spec = nullptr; in get_call_spec() local
70 def update_module_call_signatures(path, in_spec, out_spec): argument
268 const char** out_spec) { in AOTInductorModelContainerGetCallSpec()
24 DimVector out_spec = lhs_spec; in convolution_batch_rule() local
144 CompileSpec** out_spec) { in PopulateCompileSpecs()
75 ShardingSpec out_spec; member