| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| H A D | tpu_embedding_software_deduplication_rewrite_pass.cc | 70 absl::Span<const std::string> control_inputs) { in MakeRecvDeduplicationDataNodeDef()
104 absl::Span<const std::string> control_inputs) { in MakeRecvActivationsNodeDef()
168 absl::Span<const std::string> control_inputs) { in MakeSendGradientsNodeDef()
291 std::vector<Node*> control_inputs; in GetControlInputNodes() local
398 std::vector<Node*> control_inputs; in AddRecvDeduplicationDataNode() local
693 std::vector<std::string> control_inputs; member
698 std::vector<std::string> control_inputs; in GetControlInputs() local
|
| H A D | distributed_tpu_rewrite_pass.cc | 921 const std::vector<Node*>& control_inputs, in CreateXlaSplitOp()
977 const std::vector<Node*>& control_inputs, in ShardInputWithXlaSplitOp()
1092 std::vector<Node*> control_inputs; in CreateOrGetXlaSplitNodeForVariableArg() local
2489 std::vector<Node*> control_inputs; in BuildDynamicShapeNodes() local
|
| H A D | encapsulate_tpu_computations_pass.cc | 2653 gtl::FlatSet<Node*> control_inputs; in BuildTPUReplicateOps() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/framework/ |
| H A D | ops.py | 1911 control_inputs, argument
2001 control_inputs=None, argument
2536 def control_inputs(self): member in Operation
4864 def __init__(self, graph, control_inputs): argument
4918 def control_inputs(self): member in Graph._ControlDependenciesController
4988 def control_dependencies(self, control_inputs): argument
5613 def control_dependencies(control_inputs): argument
|
| H A D | func_graph.py | 469 def control_dependencies(self, control_inputs): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/graph/ |
| H A D | graph_def_builder.cc | 61 gtl::ArraySlice<Node*> control_inputs) { in WithControlInputsImpl()
|
| H A D | costmodel.cc | 484 std::vector<const Edge*> control_inputs; in AddToCostGraphDef() local
|
| H A D | testlib.cc | 303 Node* NoOp(Graph* g, const std::vector<Node*>& control_inputs) { in NoOp()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | op_selector.py | 239 control_inputs=False, argument
|
| H A D | control_flow_ops.py | 2971 def tuple_v2(tensors, control_inputs=None, name=None): argument
3002 def tuple(tensors, name=None, control_inputs=None): # pylint: disable=redefined-builtin argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | accumulate_n_optimizer.cc | 244 std::vector<Node*> control_inputs; in RewriteIntoAddN() local
|
| H A D | lower_while_op.cc | 261 std::vector<Node*> control_inputs; in CreateEnterNodes() local
|
| H A D | single_threaded_executor_test.cc | 378 std::vector<Node*> control_inputs; in BM_executor() local
|
| H A D | executor_test.cc | 460 std::vector<Node*> control_inputs; in BM_executor() local
|
| H A D | graph_constructor.cc | 866 std::set<TensorId> control_inputs; in RemapNodeDefInputs() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/nccl/ |
| H A D | nccl_rewrite.cc | 45 std::vector<Node*> control_inputs; in ReplaceReduce() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/ |
| H A D | encapsulate_xla_computations_pass.cc | 265 absl::flat_hash_set<Node*> control_inputs; in BuildXlaLaunchOps() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| H A D | breakup-islands.cc | 192 ValueRange control_inputs, in CreateIsland()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| H A D | convert_control_to_data_outputs.cc | 241 IslandOp CreateIsland(Operation* sub_op, ValueRange control_inputs, in CreateIsland()
|
| /aosp_15_r20/external/tensorflow/tensorflow/cc/framework/ |
| H A D | scope.cc | 327 std::vector<Node*> control_inputs; in UpdateBuilder() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/parallel_for/ |
| H A D | pfor.py | 333 def control_inputs(self): member in WhileOp
617 def true_fn(control_inputs, body_pfor, body_output, stacked): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/client/ |
| H A D | tf_session_helper.cc | 551 std::vector<TF_Operation*> control_inputs(TF_OperationNumControlInputs(oper)); in TF_OperationGetControlInputs_wrapper() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | function.cc | 501 std::vector<int> control_inputs; member
|
| /aosp_15_r20/external/tensorflow/tensorflow/c/ |
| H A D | c_api_test.cc | 822 TF_Operation* control_inputs[100]; in TEST() local
|
| H A D | c_api.cc | 1183 TF_Operation** control_inputs, in TF_OperationGetControlInputs()
|