Searched defs:InputKind (Results 1 – 8 of 8) sorted by relevance
40 def input(self, name: str, value: torch.Tensor, kind: InputKind):
64 class InputKind(Enum): class
63 enum InputKind { enum
72 class InputKind; variable
781 enum InputKind { IK_SUBGRAPH_INPUT, IK_OPERATION_OUTPUT, IK_VALUE }; in TEST_P() enum