Searched defs:compiled_args (Results 1 – 11 of 11) sorted by relevance
24 virtual void compiled_args(torch::dynamo::autograd::CompiledNodeArgs& args) { in compiled_args() function37 virtual void compiled_args(torch::dynamo::autograd::CompiledNodeArgs& args) { in compiled_args() function49 virtual void compiled_args(torch::dynamo::autograd::CompiledNodeArgs& args) { in compiled_args() function
176 void PyFunctionTensorPreHook::compiled_args(CompiledNodeArgs& args) { in compiled_args() function in torch::autograd::PyFunctionTensorPreHook187 void PyFunctionPreHook::compiled_args(CompiledNodeArgs& args) { in compiled_args() function in torch::autograd::PyFunctionPreHook196 void PyFunctionPostHook::compiled_args(CompiledNodeArgs& args) { in compiled_args() function in torch::autograd::PyFunctionPostHook230 void PyFunctionTensorPostAccGradHooks::compiled_args( in compiled_args() function in torch::autograd::PyFunctionTensorPostAccGradHooks
191 void compiled_args(CompiledNodeArgs& args) override { in compiled_args() function
575 virtual void compiled_args(CompiledNodeArgs& args) { in compiled_args() function
297 void PyNode::compiled_args(CompiledNodeArgs& args) { in compiled_args() function in torch::autograd::PyNode
19 void Error::compiled_args(CompiledNodeArgs& args) { in compiled_args() function in torch::autograd::Error69 void GraphRoot::compiled_args(CompiledNodeArgs& args) { in compiled_args() function in torch::autograd::GraphRoot
41 void CopyBackwards::compiled_args(CompiledNodeArgs& args) { in compiled_args() function in torch::autograd::CopyBackwards181 void CopySlices::compiled_args(CompiledNodeArgs& args) { in compiled_args() function in torch::autograd::CopySlices
68 void AccumulateGrad::compiled_args(CompiledNodeArgs& args) { in compiled_args() function in torch::autograd::AccumulateGrad
72 void compiled_args(CompiledNodeArgs& args) override {} in compiled_args() function
424 compiled_args: Vec<CompiledKernelArg>, field505 compiled_args: Vec<CompiledKernelArg>, field515 let compiled_args = CompiledKernelArg::deserialize(reader)?; in deserialize() localVariable738 let compiled_args = &mut res.compiled_args; in compile_nir_variant() localVariable993 let mut compiled_args: Vec<_> = (0..args.len()) in compile_nir_remaining() localVariable
30 void compiled_args(CompiledNodeArgs& args) override {} in compiled_args() function