| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| H A D | encapsulate_tpu_computations_pass.cc | 1745 const Node* arg_node = cond_fbody->arg_nodes[i]; in FindArgsToLiftForWhileNode() local
1777 const Node* arg_node = body_fbody->arg_nodes[i]; in FindArgsToLiftForWhileNode() local
1788 const Node* arg_node = body_fbody->arg_nodes[i]; in FindArgsToLiftForWhileNode() local
1818 const Node* arg_node = fbody.arg_nodes[i]; in FindArgsToLiftForCallNode() local
1904 Node* arg_node = fbody->arg_nodes[i]; in RemoveArgsToLiftFromFunctionBody() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/ |
| H A D | dtensor_device_util.cc | 678 tensorflow::Node* arg_node = graph->AddNode(arg_node_def, &status); in PrepareGraphForMlir() local
730 Node* arg_node = graph->AddNode(arg_node_def, &status); in PrepareGraphForMlir() local
|
| H A D | dtensor_device.cc | 1211 Node* arg_node = *it; in SelectGraphToExecute() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/ |
| H A D | graph_compiler_util.cc | 80 Node* arg_node = nullptr; in AddArgNodes() local
|
| H A D | tf2xla_util.cc | 188 Node* arg_node = arg_nodes[arg_index]; in ReplaceArgUsageWithConstNode() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/ |
| H A D | tpu_functional_ops.cc | 237 const Edge* FindHostToDeviceEdge(Node* arg_node) { in FindHostToDeviceEdge()
573 Node* arg_node = input_edge->src(); in FindTpuReplicatedInputAndXlaSharding() local
2071 Node* arg_node; in ShardInputsWithXlaSharding() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | single_threaded_executor.cc | 169 const Node* arg_node = arg_index_node_pair.second; in Initialize() local
|
| /aosp_15_r20/external/mesa3d/src/compiler/nir/ |
| H A D | nir_lower_vars_to_ssa.c | 568 struct deref_node *arg_node = get_deref_node(arg_deref, state); in lower_copies_to_load_store() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/utils/ |
| H A D | graph_partition.cc | 58 Node* arg_node = nullptr; member
|
| /aosp_15_r20/external/pytorch/torch/distributed/pipelining/ |
| H A D | stage.py | 880 def create_recv_tensor(placeholder, arg_node): argument
|
| /aosp_15_r20/external/selinux/libsepol/cil/src/ |
| H A D | cil_write_ast.c | 490 static void write_call_args_tree(FILE *out, struct cil_tree_node *arg_node) in write_call_args_tree()
|
| H A D | cil_resolve_ast.c | 2642 struct cil_tree_node *arg_node = NULL; in cil_build_call_args() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| H A D | import_model.cc | 1642 auto& arg_node = arg_nodes[i]; in ConvertFunctionArgAndRets() local
2588 Node* arg_node = arg_nodes->at(i).node; in InferMainFunctionType() local
2697 auto& arg_node = arg_node_and_idx.value(); in GetArgsRetsAndTypesFromFunctionGraph() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/ |
| H A D | extract_outside_compilation_pass.cc | 530 Node* arg_node) { in AddMatchingRetvalNode()
546 Node* arg_node) { in ReplaceLiftedArgNodePlaceholderWithArg()
|
| /aosp_15_r20/external/pytorch/test/quantization/fx/ |
| H A D | test_quantize_fx.py | 3914 def _check_node_not_observed(model, arg_node, node): argument
|