| /aosp_15_r20/external/pytorch/torch/onnx/_internal/ |
| H A D | registration.py | 139 self._functions: OverrideDict[OpsetVersion, Callable] = OverrideDict()
154 version = _dispatch_opset_version(opset, self._functions)
158 return self._functions[version]
167 if self._functions.in_base(opset):
174 self._functions.set_base(opset, func)
183 self._functions.override(opset, func)
191 if not self._functions.overridden(opset):
196 self._functions.remove_override(opset)
200 return min(self._functions)
|
| /aosp_15_r20/external/sl4a/Docs/ |
| H A D | generate_api_reference_md.py | 110 self._functions = collections.defaultdict(list)
179 self._functions[module].append(f)
197 for module in sorted(self._functions.keys()):
199 for func in self._functions[module]:
206 self._functions.values()):
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/profiler/ |
| H A D | pprof_profiler.py | 156 self._functions = functions
191 line.function_id = self._functions.index_of(
267 self._functions = Functions(self._string_table)
268 self._locations = Locations(self._functions)
357 pprof_profile.function.extend(self._functions.function_protos())
|
| /aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/UNIT/ |
| H A D | DynamicTensorFixture.h | 428 _functions.emplace_back(std::make_unique<ComplexFunctionType>(_ms.mm)); in run()
460 _functions[f]->configure(&tensors[f], &ws[f], nullptr, &tensors[f + 1], info); in run()
476 _functions[f]->run(); in run()
489 std::vector<std::unique_ptr<ComplexFunctionType>> _functions{};
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/eager/ |
| H A D | wrap_function.py | 353 for f in self.graph._functions.values():
458 self._functions = {}
462 return self._functions
553 self._functions[name] = wrapped_function
|
| H A D | lift_to_graph.py | 116 f = base_graph._functions.get(op.type, None)
117 if f is not None and compat.as_str(f.name) not in graph._functions:
|
| H A D | function_test.py | 2686 self.assertLen(graph._functions, 2)
2687 functions = list(graph._functions.values())
2731 self.assertLen(graph._functions, 6)
2733 functions = list(graph._functions.values())
2770 self.assertLen(graph._functions, 6)
2771 functions = list(graph._functions.values())
2808 self.assertLen(graph._functions, 6)
2810 functions = list(graph._functions.values())
2834 self.assertLen(graph._functions, 6)
2878 self.assertLen(graph._functions, 3)
[all …]
|
| H A D | function.py | 428 for f in self.graph._functions.values():
1371 self._functions = functions
1378 forward_function = self._functions.forward(
1389 self._functions.record(
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/data/ops/ |
| H A D | dataset_ops.py | 408 for used_function in self._functions():
428 def _functions(self): member in DatasetV2
4154 def _functions(self): member in DatasetV1Adapter
4155 return self._dataset._functions() # pylint: disable=protected-access
5421 def _functions(self): member in MapDataset
5473 def _functions(self): member in ParallelMapDataset
5505 def _functions(self): member in FlatMapDataset
5549 def _functions(self): member in InterleaveDataset
5619 def _functions(self): member in ParallelInterleaveDataset
5659 def _functions(self): member in FilterDataset
[all …]
|
| H A D | load_op.py | 80 def _functions(self): member in _LoadDataset
|
| H A D | save_op.py | 92 def _functions(self): member in _SaveDataset
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/keras/saving/saved_model/ |
| H A D | save_impl.py | 412 self._functions = weakref.WeakValueDictionary()
464 for fn in self._functions.values():
564 self._functions[name] = fn.wrapped_call
|
| /aosp_15_r20/external/pytorch/torch/autograd/ |
| H A D | function.py | 14 from torch._C import _functions
623 err_fn = _functions.DelayedError(
|
| /aosp_15_r20/external/pytorch/torch/_subclasses/ |
| H A D | meta_utils.py | 1162 r = torch._C._functions.DelayedError(
1200 r = torch._C._functions.DelayedError(
1251 r = torch._C._functions.DelayedError( # type: ignore[assignment]
1480 r = torch._C._functions.DelayedError(
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_autograd_fallback.py | 360 w = torch._C._functions.UndefinedGrad()(w)
361 z = torch._C._functions.UndefinedGrad()(z)
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/framework/ |
| H A D | convert_to_constants.py | 658 self._functions = {
675 return self._functions
718 for f in self._functions.values():
|
| H A D | ops.py | 3155 self._functions = collections.OrderedDict()
3480 for f in self._functions.values():
3537 defined_function = self._functions[function_def.signature.name]
3623 return compat.as_str(name) in self._functions
3634 return self._functions.get(compat.as_str(name), None)
3668 self._functions[compat.as_str(name)] = function
6613 memory.dismantle_ordered_dict(graph._functions) # pylint: disable=protected-access
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/data/benchmarks/ |
| H A D | from_tensor_slices_benchmark.py | 47 def _functions(self): member in SingleThreadedFlatMapDataset
|
| /aosp_15_r20/external/pytorch/torch/nn/parallel/ |
| H A D | scatter_gather.py | 6 from torch.nn.parallel._functions import Gather, Scatter
|
| H A D | replicate.py | 96 from torch.nn.parallel._functions import Broadcast
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/ops/ |
| H A D | grouping.py | 379 def _functions(self): member in _GroupByReducerDataset
|
| H A D | prefetching_ops.py | 252 def _functions(self): member in _MapOnGpuDataset
|
| H A D | batching.py | 377 def _functions(self): member in _MapAndBatchDataset
|
| /aosp_15_r20/external/pytorch/torch/nn/modules/ |
| H A D | normalization.py | 10 from ._functions import CrossMapLRN2d as _cross_map_lrn2d
|
| /aosp_15_r20/external/pytorch/torch/utils/ |
| H A D | hooks.py | 172 new_tensors = torch.nn.modules._functions.BackwardHookFunction.apply(*tensors)
|