Home
last modified time | relevance | path

Searched defs:inner_func (Results 1 – 11 of 11) sorted by relevance

/aosp_15_r20/external/pytorch/test/dynamo/
H A Dtest_ctx_manager.py1350 def inner_func(x): function
1385 def inner_func(x): function
1391 def inner_func(x): function
1419 def inner_func(x): function
1424 def inner_func(x): function
H A Dutils.py17 def inner_func(): function
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/
H A Dmesh_propagation.cc555 auto inner_func = MaybeFindFunction(&cluster.GetBody().front()); in PropagateMeshFromInputs() local
614 auto inner_func = MaybeFindFunction(op_inside_cluster); in PropagateMeshFromConsumers() local
/aosp_15_r20/external/pytorch/torch/distributed/checkpoint/
H A Dutils.py409 def inner_func(*args, **kwargs) -> Any: function
/aosp_15_r20/external/deqp-deps/SPIRV-Tools/source/opt/
Dinvocation_interlock_placement_pass.cpp153 Function* inner_func = context()->GetFunction(function_id); in recordBeginOrEndInFunction() local
/aosp_15_r20/external/angle/third_party/spirv-tools/src/source/opt/
H A Dinvocation_interlock_placement_pass.cpp153 Function* inner_func = context()->GetFunction(function_id); in recordBeginOrEndInFunction() local
/aosp_15_r20/external/swiftshader/third_party/SPIRV-Tools/source/opt/
H A Dinvocation_interlock_placement_pass.cpp153 Function* inner_func = context()->GetFunction(function_id); in recordBeginOrEndInFunction() local
/aosp_15_r20/external/tensorflow/tensorflow/python/tpu/
H A Dtpu_outside_compilation_test.py111 def inner_func(*args, **kwargs): function
/aosp_15_r20/external/python/dateutil/dateutil/
Drrule.py85 def inner_func(self, *args, **kwargs): function
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dmeta_optimizer_test.cc1291 FunctionDef inner_func = FunctionDefHelper::Create( in TEST_P() local
/aosp_15_r20/external/pytorch/test/
H A Dtest_autograd.py4198 def inner_func(x): function
4203 def inner_func(x): function