Home
last modified time | relevance | path

Searched defs:control_inputs (Results 1 – 23 of 23) sorted by relevance

/external/tensorflow/tensorflow/core/graph/
Dgraph_def_builder.cc61 gtl::ArraySlice<Node*> control_inputs) { in WithControlInputsImpl()
Dcostmodel.cc484 std::vector<const Edge*> control_inputs; in AddToCostGraphDef() local
Dtestlib.cc303 Node* NoOp(Graph* g, const std::vector<Node*>& control_inputs) { in NoOp()
/external/tensorflow/tensorflow/python/ops/
Dop_selector.py243 control_inputs=False, argument
Dcontrol_flow_ops.py3012 def tuple_v2(tensors, control_inputs=None, name=None): argument
3043 def tuple(tensors, name=None, control_inputs=None): # pylint: disable=redefined-builtin argument
/external/tensorflow/tensorflow/core/common_runtime/
Daccumulate_n_optimizer.cc244 std::vector<Node*> control_inputs; in RewriteIntoAddN() local
Dlower_while_op.cc261 std::vector<Node*> control_inputs; in CreateEnterNodes() local
Dsingle_threaded_executor_test.cc352 std::vector<Node*> control_inputs; in BM_executor() local
Dexecutor_test.cc455 std::vector<Node*> control_inputs; in BM_executor() local
Dgraph_constructor.cc864 std::set<TensorId> control_inputs; in RemapNodeDefInputs() local
/external/tensorflow/tensorflow/core/nccl/
Dnccl_rewrite.cc45 std::vector<Node*> control_inputs; in ReplaceReduce() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dbreakup-islands.cc156 ValueRange control_inputs, in CreateIsland()
/external/tensorflow/tensorflow/compiler/jit/
Dencapsulate_xla_computations_pass.cc271 absl::flat_hash_set<Node*> control_inputs; in BuildXlaLaunchOps() local
/external/tensorflow/tensorflow/python/framework/
Dops.py1841 def _create_c_op(graph, node_def, inputs, control_inputs, op_def=None): argument
1921 control_inputs=None, argument
2417 def control_inputs(self): member in Operation
4713 def __init__(self, graph, control_inputs): argument
4767 def control_inputs(self): member in Graph._ControlDependenciesController
4837 def control_dependencies(self, control_inputs): argument
5454 def control_dependencies(control_inputs): argument
Dfunc_graph.py327 def control_dependencies(self, control_inputs): argument
/external/tensorflow/tensorflow/cc/framework/
Dscope.cc327 std::vector<Node*> control_inputs; in UpdateBuilder() local
/external/tensorflow/tensorflow/python/client/
Dtf_session_helper.cc550 std::vector<TF_Operation*> control_inputs(TF_OperationNumControlInputs(oper)); in TF_OperationGetControlInputs_wrapper() local
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_rewrite_pass.cc923 const std::vector<Node*>& control_inputs, in CreateXlaSplitOp()
981 const std::vector<Node*>& control_inputs, in ShardInputWithXlaSplitOp()
1096 std::vector<Node*> control_inputs; in CreateOrGetXlaSplitNodeForVariableArg() local
2541 std::vector<Node*> control_inputs; in BuildDynamicShapeNodes() local
Dencapsulate_tpu_computations_pass.cc2701 gtl::FlatSet<Node*> control_inputs; in BuildTPUReplicateOps() local
/external/tensorflow/tensorflow/python/ops/parallel_for/
Dpfor.py337 def control_inputs(self): member in WhileOp
621 def true_fn(control_inputs, body_pfor, body_output, stacked): argument
/external/tensorflow/tensorflow/core/framework/
Dfunction.cc480 std::vector<int> control_inputs; member
/external/tensorflow/tensorflow/c/
Dc_api_test.cc822 TF_Operation* control_inputs[100]; in TEST() local
Dc_api.cc1213 TF_Operation** control_inputs, in TF_OperationGetControlInputs()