Home
last modified time | relevance | path

Searched defs:setup_context (Results 1 – 20 of 20) sorted by relevance

/external/mesa3d/src/gallium/drivers/softpipe/
Dsp_setup.c75 struct setup_context { struct
76 struct softpipe_context *softpipe;
82 const float (*vmax)[4];
83 const float (*vmid)[4];
84 const float (*vmin)[4];
85 const float (*vprovoke)[4];
87 struct edge ebot;
88 struct edge etop;
89 struct edge emaj;
91 float oneoverarea;
[all …]
/external/pytorch/torch/testing/_internal/
Dautograd_function_db.py37 def setup_context(ctx, inputs, output): member in NumpyCube
65 def setup_context(ctx, inputs, outputs): member in CubeGenVmap
93 def setup_context(ctx, inputs, output): member in NumpyCubeNotComposable
111 def setup_context(ctx, inputs, output): member in NumpyMul
157 def setup_context(ctx, inputs, outputs): member in MulGenVmap
186 def setup_context(ctx, inputs, output): member in NumpyExp_
224 def setup_context(ctx, inputs, output): member in NumpySort
262 def setup_context(ctx, inputs, outputs): member in SortGenVmap
303 def setup_context(ctx, inputs, output): member in NumpyTake
349 def setup_context(ctx, inputs, outputs): member in TakeGenVmap
[all …]
/external/pytorch/docs/source/notes/
Dextending.rst359 .. _forward-ad-autograd-function:
/external/liburing/examples/
Dlink-cp.c31 static int setup_context(unsigned entries, struct io_uring *ring) in setup_context() function
Dio_uring-cp.c30 static int setup_context(unsigned entries, struct io_uring *ring) in setup_context() function
Ducontext-cp.c93 static int setup_context(async_context *pctx, struct io_uring *ring) in setup_context() function
Dio_uring-udp.c91 static int setup_context(struct ctx *ctx) in setup_context() function
/external/pytorch/torch/_functorch/
Dautograd_function.py120 def setup_context(ctx, inputs, output): function
398 def setup_context(ctx, inputs, outputs): function
/external/pytorch/test/functorch/
Dtest_eager_transforms.py1139 def setup_context(ctx, inputs, output): member in TestAutogradFunction.test_set_materialize_grads.A
1171 def setup_context(ctx, inputs, output): member in TestAutogradFunction.test_function_returns_input.A
1249 def setup_context(ctx, inputs, output): member in TestAutogradFunction.test_needs_input_grads.A
1272 def setup_context(ctx, inputs, output): member in TestAutogradFunction._get_NumpyCubeNotComposable.NumpyCubeNotComposable
1327 def setup_context(ctx, inputs, output): member in TestAutogradFunction.test_grad_fn_name.FooBar
1355 def setup_context(ctx, inputs, output): member in TestAutogradFunctionVmapAPI.test_no_vmap_staticmethod_and_no_generate_vmap_rule.NumpyCube
1377 def setup_context(ctx, outputs, input): member in TestAutogradFunctionVmapAPI.test_has_vmap_staticmethod_and_has_generate_vmap_rule.NumpyCube
1401 def setup_context(ctx, inputs, output): member in TestAutogradFunctionVmapAPI.test_info_object.Id
1426 def setup_context(ctx, inputs, output): member in TestAutogradFunctionVmapAPI.test_in_dims_single_input.Id
1450 def setup_context(ctx, inputs, output): member in TestAutogradFunctionVmapAPI.test_in_dims_multiple_inputs.Id
[all …]
Dtest_aotdispatch.py2431 def setup_context(ctx, inputs, output): function
2489 def setup_context(ctx, inputs, output): function
/external/pytorch/test/
Dtest_flop_counter.py169 def setup_context(ctx, inputs, output): member in TestFlopCounter.test_conv_backwards_as_decomposition.onlyConvs
Dtest_custom_ops.py2246 def setup_context(ctx, inputs, output): function
2353 def setup_context(ctx, inputs, keyword_only_inputs, output): function
2383 def setup_context(ctx, inputs, keyword_only_inputs, output): function
2968 def setup_context(ctx, inputs, output) -> Tensor: function
3012 def setup_context(ctx, inputs, output) -> Tensor: function
Dtest_autograd.py688 def setup_context(ctx, inputs, output): member in TestAutograd.test_custom_function_vmap_defaults.MySquare
707 def setup_context(ctx, inputs, output): member in TestAutograd.test_custom_function_setup_context_simple.MySquare
730 def setup_context(ctx, inputs, output): member in TestAutograd.test_custom_function_setup_context_multi_output.MySquare
752 def setup_context(ctx, inputs, output): member in TestAutograd.test_custom_function_setup_context_multi_input.MyReshape
10236 def setup_context(ctx, inputs, output): member in TestAutograd.test_setup_context_when_forward_has_default_args.PowFunction
10253 def setup_context(cls, ctx, inputs, output): member in TestAutograd.test_setup_context_when_forward_has_default_args.PowFunctionWithClassmethod
13802 def setup_context(ctx, inputs, output) -> torch.Tensor: function
/external/pytorch/torch/autograd/
Dfunction.py392 def setup_context(ctx: Any, inputs: Tuple[Any, ...], output: Any) -> Any: member in _SingleLevelFunction
/external/pytorch/torch/utils/
Dcheckpoint.py784 def setup_context(ctx: Any, inputs: Tuple[Any, ...], output: Any) -> None: member in _NoopSaveInputs
/external/python/cpython3/Python/
D_warnings.c851 setup_context(Py_ssize_t stack_level, in setup_context() function
/external/pytorch/torch/csrc/autograd/
Dpython_function.cpp1270 auto setup_context = PyObject_GetAttrString(function, "setup_context"); in get_base_setup_context() local
/external/pytorch/test/nn/
Dtest_module_hooks.py422 def setup_context(): function
/external/pytorch/test/inductor/
Dtest_compiled_autograd.py186 def setup_context(ctx, inputs, output): function
/external/pytorch/test/dynamo/
Dtest_autograd_function.py100 def setup_context(ctx, inputs, output): member in LinearFunction