| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| H A D | encapsulate_tpu_computations_pass.cc | 293 int dst_input = e->dst_input(); in RemoveIdentityNodesForArgRetval() local
385 int dst_input = e->dst_input(); in MoveHeadOutsideCompilationToHost() local
645 int dst_input = original_edge->dst_input(); in MoveHeadOutsideCompilationToHost() local
830 int dst_input = i * num_new_per_replica_inputs + iter->second; in RemoveUnusedXlaInput() local
851 int dst_input = num_replicas * num_new_per_replica_inputs + iter->second - in RemoveUnusedXlaInput() local
1073 int dst_input = node_input_pair.second; in MoveTailOutsideCompilationToHost() local
1308 int dst_input = e->dst_input(); in ReplaceArgUsedByOutsideCompilationWithPlaceholder() local
1409 int dst_input = e->dst_input(); in RemoveUnusedXlaOutput() local
1497 int dst_input = output_edge->dst_input(); in RemoveEdgesBetweenArgAndRetval() local
1518 int dst_input = e->dst_input(); in RemoveEdgesBetweenArgAndRetval() local
[all …]
|
| H A D | distributed_tpu_rewrite_helpers.h | 83 int dst_input; member
|
| H A D | distributed_tpu_rewrite_pass.cc | 573 int dst_input = edge->dst_input(); in ReplaceCompilationResultNodeWithIdentity() local
1161 int dst_input = 1; in CreateConcatNode() local
3796 int dst_input = edge->dst_input(); in BuildExecuteNodes() local
3820 int dst_input = replicate_edge_to_replace->dst_input(); in BuildExecuteNodes() local
4060 int dst_input = edge->dst_input(); in CopyOutsideCompilationEdges() local
4082 int dst_input = edge->dst_input(); in CopyOutsideCompilationEdges() local
4342 int dst_input = e->dst_input(); in LowerOutsideCompilationFunctionalNodes() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/ |
| H A D | rearrange_function_argument.cc | 120 int dst_input = e->dst_input(); in ReorderInputEdges() local
146 int dst_input = e->dst_input(); in ReorderOutputEdges() local
239 int dst_input = e->dst_input(); in RearrangeOutputEdges() local
418 int dst_input = e->dst_input(); in MaybeRewriteIfNode() local
482 int dst_input = e->dst_input(); in MaybeRewriteIfNode() local
|
| H A D | functionalize_while.cc | 254 int dst_input = edges[i]->dst_input(); in FunctionalizeLoop() local
479 int dst_input = edge->dst_input(); in FunctionalizeLoop() local
|
| H A D | functionalize_cond.cc | 467 int dst_input = IsMerge(e->dst()) ? 0 : e->dst_input(); in BuildArgumentNodes() local
521 int dst_input = edge->dst_input(); in AddSwitchNodeAlongEdge() local
888 int dst_input = edge->dst_input(); in AddOutputEdges() local
1287 int dst_input = e->dst_input(); in RemoveRedundantSwitch() local
|
| H A D | tf2xla_util.cc | 115 StatusOr<Node*> ReplaceEdge(Graph* g, Node* dst, int dst_input, Node* with, in ReplaceEdge()
137 int dst_node_id, dst_input; in ReplaceSrcOutputUsageWithNode() member
1098 int dst_input = e->dst_input(); in RewriteTensorListWithConstElement() local
|
| H A D | tf2xla_util.h | 174 int src_output, dst_input; member
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/ |
| H A D | encapsulate_util.cc | 119 int dst_input, dst_node_id; in PreprocessDataEdgesBetweenOutsideCompilations() member
147 int src_output = e->src_output(), dst_input = e->dst_input(); in PreprocessDataEdgesBetweenOutsideCompilations() local
242 int dst_input = data_edges[i].dst_input; in PostprocessDataEdgesBetweenOutsideCompilations() local
|
| H A D | shape_inference_helpers.h | 39 int dst_input; member
|
| H A D | clone_constants_for_better_clustering.cc | 147 int dst_input = e->dst_input(); in CloneSmallHostConstantInputs() local
|
| H A D | shape_inference.cc | 154 int dst_input = e->dst_input(); in PropagateShapes() local
|
| H A D | increase_dynamism_for_auto_jit_pass.cc | 290 int dst_input = e->dst_input(); in ReplaceTensorFlowSliceWithStaticShapedSlice() local
|
| H A D | encapsulate_subgraphs_pass.cc | 914 int dst_input = FindOutputSlotOfEdgeDst(src_func_id, dst_func_id, edge); in CopyEdgeToOutputGraph() local
|
| H A D | extract_outside_compilation_pass.cc | 2191 int dst_input = e->dst_input(); in CopyOutsideCompilationConstNodes() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | placer_inspection_required_ops_utils.cc | 236 int dst_input = edge->dst_input(); in AddOutputIdentities() local
|
| H A D | shape_refiner.cc | 323 int dst_input = e->dst_input(); in UpdateNode() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/graph/ |
| H A D | graph.h | 437 int dst_input() const { return dst_input_; } in dst_input() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/ |
| H A D | graph_properties.cc | 1003 for (int dst_input = 0; dst_input < ic->num_inputs(); ++dst_input) { in UpdateNode() local
1116 for (int dst_input = 0; dst_input < ic->num_inputs(); ++dst_input) { in UpdateNode() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/ |
| H A D | segment.cc | 74 int dst_input() const { return dst_port_; } in dst_input() function in tensorflow::tensorrt::segment::__anon2febd5fa0111::SimpleEdge
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| H A D | import_model.cc | 2236 int dst_input) { in AddBackedge()
|