Home
last modified time | relevance | path

Searched defs:src_output (Results 1 – 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_rewrite_helpers.h81 int src_output; member
Dencapsulate_tpu_computations_pass.cc600 int src_output = e->src_output(); in MoveHeadOutsideCompilationToHost() local
624 int src_output = in MoveHeadOutsideCompilationToHost() local
860 int src_output = e->src_output(); in RemoveUnusedXlaInput() local
881 int src_output = e->src_output(); in RemoveUnusedXlaInput() local
1175 int src_output = replica_id * new_num_replicated_outputs + in MoveTailOutsideCompilationToHost() local
1191 int src_output = original_edge->src_output(); in MoveTailOutsideCompilationToHost() local
1451 int src_output = in RemoveUnusedXlaOutput() local
Ddistributed_tpu_rewrite_pass.cc552 int src_output = edge->src_output(); in ReplaceCompilationResultNodeWithIdentity() local
2257 int src_output; in BuildDynamicShapeNodes() local
3884 int src_output = e->src_output(); in LowerOutsideCompilationFunctionalNodes() local
/external/tensorflow/tensorflow/compiler/tf2xla/
Drearrange_function_argument.cc119 int src_output = e->src_output(); in ReorderInputEdges() local
143 int src_output = e->src_output(); in ReorderOutputEdges() local
240 int src_output = e->src_output(); in RearrangeOutputEdges() local
417 int src_output = e->src_output(); in MaybeRewriteIfNode() local
483 int src_output = e->src_output(); in MaybeRewriteIfNode() local
Dtf2xla_util.h176 int src_output, dst_input; member
Dfunctionalize_while.cc101 int src_output = squash_src_outputs[e->src()->id()] && !e->IsControlEdge() in CopySubgraph() local
Dfunctionalize_cond.cc211 int src_output; member
512 int src_output = edge->src_output(); in AddSwitchNodeAlongEdge() local
702 int src_output = e->src_output(); in ExtractBodies() local
/external/tensorflow/tensorflow/compiler/jit/
Dshape_inference_helpers.h37 int src_output; member
Dincrease_dynamism_for_auto_jit_pass.cc289 int src_output = e->src_output(); in ReplaceTensorFlowSliceWithStaticShapedSlice() local
Dencapsulate_util.cc147 int src_output = e->src_output(), dst_input = e->dst_input(); in PreprocessDataEdgesBetweenOutsideCompilations() local
Dencapsulate_subgraphs_pass.cc938 int src_output = FindOutputSlotOfEdgeSrc(src_func_id, dst_func_id, edge); in CopyEdgeToOutputGraph() local
/external/tensorflow/tensorflow/core/common_runtime/
Dplacer_inspection_required_ops_utils.cc214 Node** identity_node) { in AddOutputIdentities()
249 int src_output = edge->src_output(); in AddOutputIdentities() local
Dlower_case_op.cc146 Status CaseBuilder::AddInput(Node* src, int src_output) { in AddInput()
Dlower_if_op.cc179 Status CondBuilder::AddInput(Node* src, int src_output) { in AddInput()
Dshape_refiner.cc308 int src_output = e->src_output(); in UpdateNode() local
/external/tensorflow/tensorflow/core/debug/
Ddebug_graph_utils.cc137 const int src_output = edge->src_output(); in InsertNodes() local
301 const string& src_node_name, const int src_output, const DataType src_dt, in CreateCopyNode()
/external/tensorflow/tensorflow/core/graph/
Dgraph.h412 int src_output() const { return src_output_; } in src_output() function
/external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/
Dsegment.cc69 int src_output() const { return src_port_; } in src_output() function in tensorflow::tensorrt::segment::__anon3618f86b0111::SimpleEdge
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties.cc977 int src_output = fanin.port_id; in UpdateNode() local