Searched refs:create_subclass_meta (Results 1 – 3 of 3) sorted by relevance
49 from .subclass_utils import create_subclass_meta739 subclass_inp_meta=create_subclass_meta(flat_args),740 subclass_fw_graph_out_meta=create_subclass_meta(fw_graph_outs),741 subclass_tangent_meta=create_subclass_meta(traced_tangents),
59 create_subclass_meta,776 subclass_meta.grad_input_metas = create_subclass_meta(grad_inputs)
84 def create_subclass_meta( function