| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_helpers.h | 83 int dst_input; member
|
| D | encapsulate_tpu_computations_pass.cc | 319 int dst_input = e->dst_input(); in RemoveIdentityNodesForArgRetval() local
411 int dst_input = e->dst_input(); in MoveHeadOutsideCompilationToHost() local
675 int dst_input = original_edge->dst_input(); in MoveHeadOutsideCompilationToHost() local
861 int dst_input = i * num_new_per_replica_inputs + iter->second; in RemoveUnusedXlaInput() local
882 int dst_input = num_replicas * num_new_per_replica_inputs + iter->second - in RemoveUnusedXlaInput() local
1104 int dst_input = node_input_pair.second; in MoveTailOutsideCompilationToHost() local
1349 int dst_input = e->dst_input(); in ReplaceArgUsedByOutsideCompilationWithPlaceholder() local
1450 int dst_input = e->dst_input(); in RemoveUnusedXlaOutput() local
1538 int dst_input = output_edge->dst_input(); in RemoveEdgesBetweenArgAndRetval() local
1560 int dst_input = e->dst_input(); in RemoveEdgesBetweenArgAndRetval() local
[all …]
|
| D | distributed_tpu_rewrite_pass.cc | 553 int dst_input = edge->dst_input(); in ReplaceCompilationResultNodeWithIdentity() local
945 int dst_input = 1; in CreateConcatNode() local
3350 int dst_input = edge->dst_input(); in BuildExecuteNodes() local
3374 int dst_input = replicate_edge_to_replace->dst_input(); in BuildExecuteNodes() local
3606 int dst_input = edge->dst_input(); in CopyOutsideCompilationEdges() local
3628 int dst_input = edge->dst_input(); in CopyOutsideCompilationEdges() local
3888 int dst_input = e->dst_input(); in LowerOutsideCompilationFunctionalNodes() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| 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
|
| D | tf2xla_util.h | 176 int src_output, dst_input; member
|
| D | functionalize_while.cc | 255 int dst_input = edges[i]->dst_input(); in FunctionalizeLoop() local
483 int dst_input = edge->dst_input(); in FunctionalizeLoop() local
|
| D | functionalize_cond.cc | 469 int dst_input = IsMerge(e->dst()) ? 0 : e->dst_input(); in BuildArgumentNodes() local
523 int dst_input = edge->dst_input(); in AddSwitchNodeAlongEdge() local
893 int dst_input = edge->dst_input(); in AddOutputEdges() local
1294 int dst_input = e->dst_input(); in RemoveRedundantSwitch() local
|
| D | tf2xla_util.cc | 136 int dst_node_id, dst_input; in ReplaceArgUsageWithConstNode() member
911 int dst_input = e->dst_input(); in RewriteTensorListWithConstElement() local
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | shape_inference_helpers.h | 39 int dst_input; member
|
| 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
244 int dst_input = data_edges[i].dst_input; in PostprocessDataEdgesBetweenOutsideCompilations() local
|
| D | clone_constants_for_better_clustering.cc | 124 int dst_input = e->dst_input(); in CloneSmallHostConstantInputs() local
|
| D | shape_inference.cc | 157 int dst_input = e->dst_input(); in PropagateShapes() local
|
| D | increase_dynamism_for_auto_jit_pass.cc | 290 int dst_input = e->dst_input(); in ReplaceTensorFlowSliceWithStaticShapedSlice() local
|
| D | encapsulate_subgraphs_pass.cc | 940 int dst_input = FindOutputSlotOfEdgeDst(src_func_id, dst_func_id, edge); in CopyEdgeToOutputGraph() local
|
| D | extract_outside_compilation_pass.cc | 2190 int dst_input = e->dst_input(); in CopyOutsideCompilationConstNodes() local
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | placer_inspection_required_ops_utils.cc | 248 int dst_input = edge->dst_input(); in AddOutputIdentities() local
|
| D | shape_refiner.cc | 307 int dst_input = e->dst_input(); in UpdateNode() local
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | graph.h | 417 int dst_input() const { return dst_input_; } in dst_input() function
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/ |
| D | segment.cc | 70 int dst_input() const { return dst_port_; } in dst_input() function in tensorflow::tensorrt::segment::__anon3618f86b0111::SimpleEdge
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties.cc | 974 for (int dst_input = 0; dst_input < ic->num_inputs(); ++dst_input) { in UpdateNode() local
1087 for (int dst_input = 0; dst_input < ic->num_inputs(); ++dst_input) { in UpdateNode() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | remote_fused_graph_execute_utils.cc | 985 const int dst_input = edge->dst_input(); in FuseCluster() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | import_model.cc | 2055 int dst_input) { in AddBackedge()
|