Searched refs:enclosing_graph (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/framework/ |
| H A D | convert_to_constants.py | 84 def __init__(self, enclosing_graph): argument
85 self._enclosing_graph = enclosing_graph
150 def __init__(self, function, enclosing_graph): argument
151 super(_Function, self).__init__(enclosing_graph)
155 _Node.new(node=n, function=self, enclosing_graph=enclosing_graph)
222 def __init__(self, node, function, enclosing_graph): argument
223 super(_Node, self).__init__(enclosing_graph)
231 def new(node, function, enclosing_graph): argument
234 return _VarHandle(node, function, enclosing_graph)
236 return _Case(node, function, enclosing_graph)
[all …]
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/ |
| H A D | symbolic_shape_runtime_fusion.cpp | 26 auto enclosing_graph = tensorexpr_graph->owningGraph(); in InsertSymbolicShapesCompute() local
48 shape_compute_graph_inputs.push_back(enclosing_graph->insert( in InsertSymbolicShapesCompute()
55 *enclosing_graph, in InsertSymbolicShapesCompute()
|