Searched refs:src_slot (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | encapsulate_subgraphs_pass.cc | 774 int src_slot = edge->src_output(); in RecordArg() local
778 OutputTensor(src_node, src_slot), args_by_src_.size()); in RecordArg()
783 absl::StrCat(src_node->name(), "_", src_slot, "_arg"), kArgOp, in RecordArg()
810 int src_slot = edge->src_output(); in RecordResult() local
814 results_.emplace(OutputTensor(src_node, src_slot), results_.size()); in RecordResult()
819 absl::StrCat(src_node->name(), "_", src_slot, "_retval"), kRetValOp, in RecordResult()
821 DataType dtype = src_node->output_type(src_slot); in RecordResult()
824 builder.Input(src_image->name(), src_slot, dtype); in RecordResult()
830 graph_->AddEdge(src_image, src_slot, ret, 0); in RecordResult()
956 int src_slot = input_src.first.index; in AddHostComputes() local
[all …]
|
| /external/tensorflow/tensorflow/core/util/ |
| D | presized_cuckoo_map.h | 288 inline void CopyItem(uint64 src_bucket, int src_slot, uint64 dst_bucket, in CopyItem() argument
292 dst_ref.keys[dst_slot] = src_ref.keys[src_slot]; in CopyItem()
293 dst_ref.values[dst_slot] = src_ref.values[src_slot]; in CopyItem()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | graph.cc | 612 void AddInput(NodeDef* dst, StringPiece src_name, int src_slot) { in AddInput() argument
613 if (src_slot == Graph::kControlSlot) { in AddInput()
615 } else if (src_slot == 0) { in AddInput()
618 dst->add_input(strings::StrCat(src_name, ":", src_slot)); in AddInput()
|
| D | graph_partition.cc | 160 void AddInput(NodeDef* dst, StringPiece src_name, int src_slot) { in AddInput() argument
161 if (src_slot == Graph::kControlSlot) { in AddInput()
163 } else if (src_slot == 0) { in AddInput()
166 dst->add_input(strings::StrCat(src_name, ":", src_slot)); in AddInput()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | executor.cc | 2676 const int src_slot = e.output_slot; in ActivateNodes() local
2687 const bool is_control_edge = (src_slot == Graph::kControlSlot); in ActivateNodes()
2701 if ((*outputs)[src_slot].has_value) { in ActivateNodes()
2728 (is_dead || (!is_control_edge && !(*outputs)[src_slot].has_value)); in ActivateNodes()
2740 input_tensors[dst_loc] = std::move((*outputs)[src_slot]); in ActivateNodes()
2742 input_tensors[dst_loc] = (*outputs)[src_slot]; in ActivateNodes()
|