Home
last modified time | relevance | path

Searched defs:input_edge (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dparallel_concat_optimizer.cc78 for (const Edge* input_edge : n->in_edges()) { in Run() local
Daccumulate_n_optimizer.cc132 for (const Edge* input_edge : n->in_edges()) { in RewriteIntoTempVariable() local
Dshape_refiner.cc398 const Edge* input_edge; in EvaluateConstantTensorForEdge() local
436 const Edge* input_edge; in ConstantPartialShape() local
/external/tensorflow/tensorflow/compiler/tf2xla/
Drearrange_function_argument.cc152 const Edge* input_edge; in ReorderOutputEdges() local
246 const Edge* input_edge; in RearrangeOutputEdges() local
489 const Edge* input_edge; in MaybeRewriteIfNode() local
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dencapsulate_tpu_computations_pass.cc305 const Edge* input_edge; in RemoveIdentityNodesForArgRetval() local
719 const Edge* input_edge = in MoveHeadOutsideCompilationToHost() local
1533 const Edge* input_edge = input_edges.at(input_index); in RemoveEdgesBetweenArgAndRetval() local
2079 const Edge* input_edge; in ReplaceOutputEdgesWithInputEdgeSourceForWhile() local
Ddistributed_tpu_rewrite_pass.cc3885 const Edge* input_edge; in LowerOutsideCompilationFunctionalNodes() local
/external/tensorflow/tensorflow/core/graph/
Dgraph.cc244 Status Node::input_edge(int idx, const Edge** e) const { in input_edge() function in tensorflow::Node
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dimport_model.cc1916 for (const auto* input_edge : in_edges) { in ConvertNode() local