Searched refs:WrappedFunction (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/eager/ |
| H A D | wrap_function.py | 217 class WrappedFunction(function.ConcreteFunction): class
228 super(WrappedFunction, self).__init__(fn_graph, attrs=attrs)
243 return super(WrappedFunction, self)._call_impl(
372 pruned_fn = WrappedFunction(
457 self._wrapped_function = WrappedFunction(self.graph, self._variable_holder)
618 return WrappedFunction(
|
| /aosp_15_r20/external/pytorch/torch/_inductor/ |
| H A D | cudagraph_utils.py | 45 class WrappedFunction: class
124 func: WrappedFunction, argument
|
| H A D | cudagraph_trees.py | 94 WrappedFunction,
591 wrapped_function: WrappedFunction, argument
774 wrapped_function: WrappedFunction, argument
1825 self.ids_to_funcs: Dict[FunctionID, WrappedFunction] = {}
2200 self.ids_to_funcs[id] = WrappedFunction(
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/python/ |
| H A D | quantize_model.py | 405 func: wrap_function.WrappedFunction, argument
|