/external/tensorflow/tensorflow/core/graph/ |
D | tensor_id.h | 33 struct TensorId : public std::pair<StringPiece, int> { struct 41 TensorId() : Base() {} in TensorId() argument 42 TensorId(const SafeTensorId& id); 54 std::size_t operator()(const TensorId& x) const { in operator() argument 60 TensorId ParseTensorName(const string& name); 61 TensorId ParseTensorName(StringPiece name); 63 bool IsTensorIdControl(const TensorId& tensor_id); 74 SafeTensorId(const TensorId& id); 86 std::size_t operator()(const TensorId& x) const { in operator()
|
D | tensor_id.cc | 25 TensorId::TensorId(const SafeTensorId& id) : TensorId(id.first, id.second) {} in TensorId() function in tensorflow::TensorId 27 SafeTensorId::SafeTensorId(const TensorId& id) in SafeTensorId() 30 TensorId ParseTensorName(const string& name) { in ParseTensorName() 34 TensorId ParseTensorName(StringPiece name) { in ParseTensorName() 50 TensorId id; in ParseTensorName() 65 bool IsTensorIdControl(const TensorId& tensor_id) { in IsTensorIdControl()
|
D | tensor_id_test.cc | 82 TensorId id; in BM_ParseTensorName() 95 TensorId tensor_id = ParseTensorName(input); in TEST()
|
D | subgraph.cc | 64 TensorId id(ParseTensorName(t)); in FeedInputs() 132 TensorId id(ParseTensorName(t)); in FetchOutputs() 179 TensorId id = ParseTensorName(node_or_tensor_name); in AddNodeToTargets()
|
/external/tensorflow/tensorflow/core/common_runtime/ |
D | graph_constructor_test.cc | 1184 opts.input_map[TensorId("new_input", 0)] = TensorId("input", 1); in TEST_F() 1185 opts.input_map[TensorId("new_input", 1)] = TensorId("input", 0); in TEST_F() 1226 opts.input_map[TensorId("input", 0)] = TensorId("input", 0); in TEST_F() 1227 opts.input_map[TensorId("input", 1)] = TensorId("input", 0); in TEST_F() 1288 opts.input_map[TensorId("W2", kControlSlot)] = TensorId("W1", kControlSlot); in TEST_F() 1289 opts.input_map[TensorId("W3", kControlSlot)] = TensorId("W1", kControlSlot); in TEST_F() 1325 opts.input_map[TensorId("W1", kControlSlot)] = TensorId("W1", kControlSlot); in TEST_F() 1352 opts.input_map[TensorId("W2", Graph::kControlSlot)] = TensorId("W1", 0); in TEST_F() 1364 opts.input_map[TensorId("W2", 0)] = TensorId("W1", Graph::kControlSlot); in TEST_F() 1384 opts.input_map[TensorId("input2", 0)] = TensorId("input1", 3); in TEST_F() [all …]
|
D | graph_constructor.cc | 126 std::map<TensorId, TensorId> input_map; 129 std::vector<TensorId> return_tensors; 314 std::set<TensorId> used_input_map_keys_; 552 bool NodeNameInValues(const std::map<TensorId, TensorId>& input_map, in NodeNameInValues() argument 620 TensorId src = mapping.first; in ValidateInputMapAndControlDependencies() 621 TensorId dst = mapping.second; in ValidateInputMapAndControlDependencies() 718 TensorId id(ParseTensorName(input_name)); in InitFromEdges() 731 TensorId id(ParseTensorName(input_name)); in InitFromEdges() 870 std::set<TensorId> control_inputs; in RemapNodeDefInputs() 878 TensorId new_input = iter->second; in RemapNodeDefInputs() [all …]
|
D | session_state.cc | 76 TensorId id(ParseTensorName(name)); in SaveTensors()
|
/external/tensorflow/tensorflow/core/grappler/ |
D | mutable_graph_view.h | 141 Status AddRegularFanin(absl::string_view node_name, const TensorId& fanin); 152 const TensorId& fanin); 176 const TensorId& fanin); 185 Status RemoveRegularFanin(absl::string_view node_name, const TensorId& fanin); 219 Status UpdateFanin(absl::string_view node_name, const TensorId& from_fanin, 220 const TensorId& to_fanin); 230 const TensorId& fanin);
|
D | mutable_graph_view.cc | 43 bool IsTensorIdPortValid(const TensorId& tensor_id) { in IsTensorIdPortValid() 47 bool IsTensorIdRegular(const TensorId& tensor_id) { in IsTensorIdRegular() 51 bool IsTensorIdControlling(const TensorId& tensor_id) { in IsTensorIdControlling() 65 TensorId tensor_id = ParseTensorName(node.input(0)); in IsIdentityConsumingSwitch() 123 TensorId tensor_id = ParseTensorName(control_fanout.node->input(i)); in SwapControlledFanoutInputs() 267 Status CheckFaninIsRegular(const TensorId& fanin, ErrorHandler handler) { in CheckFaninIsRegular() 275 Status CheckFaninIsValid(const TensorId& fanin, ErrorHandler handler) { in CheckFaninIsValid() 284 const TensorId& fanin, ErrorHandler handler) { in CheckAddingFaninToSelf() 293 const TensorId& fanin, ErrorHandler handler) { in CheckRemovingFaninFromSelf() 349 TensorId tensor_id = ParseTensorName(node->input(pos)); in AddAndDedupFanouts() [all …]
|
D | utils.cc | 91 string TensorIdToString(const TensorId& tensor_id) { in TensorIdToString() 102 TensorId tensor1 = ParseTensorName(name1); in IsSameInput() 103 TensorId tensor2 = ParseTensorName(name2); in IsSameInput() 111 bool IsControlInput(const TensorId& tensor_id) { return tensor_id.index() < 0; } in IsControlInput() 234 TensorId tensor = ParseTensorName(node_as_input); in HasRegularOutputs() 249 TensorId tensor = ParseTensorName(node_as_input); in HasControlOutputs() 265 TensorId tensor = ParseTensorName(node_as_input); in NumControlOutputs() 284 const TensorId tensor = ParseTensorName(node_as_input); in NumNonControlOutputs()
|
D | graph_view.h | 159 TensorId tensor_id = ParseTensorName(port.node->input(i)); in GetFanin() 177 TensorId tensor_id = ParseTensorName(port.node->input(port.port_id)); in GetRegularFanin() 182 bool HasFanin(const NodeDefT& node, const TensorId& fanin) const { in HasFanin() 236 TensorId tensor_id = ParseTensorName(node.input(i)); in GetFanins() 307 TensorId tensor_id = ParseTensorName(node.input(i)); in GetFaninEdges() 340 TensorId tensor_id = ParseTensorName(node->input(i)); in AddFanouts()
|
D | utils.h | 58 TensorId id = ParseTensorName(input_name); in NodePositionIfSameNode() 68 TensorId id = ParseTensorName(name); in ParseNodeNameAsStringPiece() 280 string TensorIdToString(const TensorId& tensor_id); 291 bool IsControlInput(const TensorId& tensor_id);
|
D | mutable_graph_view_test.cc | 40 TensorId tensor_id = ParseTensorName(fanins[i]); in CompareNodeFanins() 62 TensorId tensor_id = ParseTensorName(fanout); in CompareNodeFanouts() 909 const TensorId& fanin_to_add, bool success, in TestAddRegularFanin() 1041 int port, const TensorId& fanin_to_add, in TestAddRegularFaninByPort() 1184 void CheckFanoutRemoved(const MutableGraphView& graph, const TensorId& fanin, in CheckFanoutRemoved() 1195 const TensorId& fanin_to_remove, bool success, in TestRemoveRegularFanin() 1467 TensorId tensor_id; in TestRemoveAllFanins() 1538 const TensorId& from_fanin, const TensorId& to_fanin, in TestUpdateFanin() 1681 void TestUpdateFaninFromFaninToNodeAsSwitchControl(const TensorId& fanin) { in TestUpdateFaninFromFaninToNodeAsSwitchControl() 1716 int port, const TensorId& fanin, bool success, in TestUpdateRegularFaninByPort() [all …]
|
/external/tensorflow/tensorflow/tools/graph_transforms/ |
D | fold_constants_lib.cc | 53 StringPieceMap<TensorId> recv_node_map; in ReplaceSendRecvs() 60 TensorId id = ParseTensorName(inputs[i]); in ReplaceSendRecvs() 84 TensorId id = ParseTensorName(input); in ReplaceSendRecvs() 152 std::unordered_set<TensorId, TensorId::Hasher> input_names; in RemoveUnusedNodes() 159 TensorId id = ParseTensorName(name); in RemoveUnusedNodes() 174 TensorId id = ParseTensorName(input); in RemoveUnusedNodes() 185 for (const TensorId& id : input_names) { in RemoveUnusedNodes() 242 std::vector<TensorId> input_names; in FoldConstants() 248 const auto compare = [](TensorId lhs, TensorId rhs) { in FoldConstants() 279 TensorId{node->name(), 0}, compare); in FoldConstants()
|
/external/tensorflow/tensorflow/compiler/jit/ |
D | deadness_analysis.cc | 283 explicit SymbolPredicate(int64 id, TensorId tensor_id, bool must_be_true) in SymbolPredicate() 301 TensorId tensor_id() const { return tensor_id_; } in tensor_id() 305 TensorId tensor_id_; 316 explicit IntSymbolPredicate(int64 id, TensorId tensor_id, in IntSymbolPredicate() 338 TensorId tensor_id() const { return tensor_id_; } in tensor_id() 344 TensorId tensor_id_; 420 TensorId tensor_id(node->name(), output_idx); in MakeSymbolPredicate() 456 TensorId tensor_id(node->name(), output_idx); in MakeSymbolPredicate() 834 absl::flat_hash_map<TensorId, string, TensorId::Hasher> PredicateMapAsString() 849 predicate_map_.insert({TensorId(n->name(), output_idx), pred}); in SetPredicate() [all …]
|
D | deadness_analysis_internal.h | 27 using PredicateMapTy = absl::flat_hash_map<TensorId, string, TensorId::Hasher>;
|
D | introduce_floating_point_jitter_pass.cc | 37 TensorId tensor_id = ParseTensorName(tensor_name); in GetNodesToModify() 70 VLOG(1) << "No users for " << TensorId(n->name(), oidx).ToString(); in IntroduceJitterToTensor() 75 << TensorId(n->name(), oidx).ToString(); in IntroduceJitterToTensor()
|
/external/tensorflow/tensorflow/tools/android/inference_interface/java/org/tensorflow/contrib/android/ |
D | TensorFlowInferenceInterface.java | 208 TensorId tid = TensorId.parse(o); in run() 577 TensorId tid = TensorId.parse(inputName); in addFeed() 583 private static class TensorId { class in TensorFlowInferenceInterface 590 public static TensorId parse(String name) { in parse() 591 TensorId tid = new TensorId(); in parse()
|
/external/tensorflow/tensorflow/compiler/tf2xla/ |
D | tf2xla.proto | 13 // TensorId identifies a tensor in a TensorFlow graph, by specifying the output 17 message TensorId { message 25 TensorId id = 1; 39 TensorId id = 1;
|
D | tf2xla_util.h | 51 string TensorIdToString(const tf2xla::TensorId& id);
|
/external/tensorflow/tensorflow/core/kernels/ |
D | remote_fused_graph_execute_utils.cc | 49 const TensorId tid = ParseTensorName(node_name_and_port); in BuildNodeSetFromNodeNamesAndPorts() 66 const TensorId tid = ParseTensorName(input); in FindNodeDefByName() 77 TensorId* tid) { in IsSameNodeName() 88 const TensorId tid0 = ParseTensorName(tensor_name); in ContainsSameTensorId() 90 const TensorId tid1 = ParseTensorName(name); in ContainsSameTensorId() 328 const TensorId tid = ParseTensorName(pair.first); in IsInputNode() 425 const TensorId tid = ParseTensorName(name_and_port); in GetOutputTensorShapeType() 525 const TensorId tid = ParseTensorName(node_name); in GetTensorShapeType() 570 const TensorId tid = ParseTensorName(name); in EmplaceTensorShapeType() 694 const TensorId tid = ParseTensorName(input); in BuildRemoteFusedGraphExecuteOpNode() [all …]
|
/external/tensorflow/tensorflow/core/grappler/utils/ |
D | graph_view.h | 298 const TensorId& fanin); 302 const TensorId& fanin); 421 Status CheckFaninsInternal(std::vector<std::vector<TensorId>>* fanins); 423 void AddFaninsInternal(std::vector<std::vector<TensorId>>* fanins);
|
D | graph_view.cc | 127 TensorId fanin_id = ParseTensorName(input); in CheckAndAddFaninsInternal() 218 bool IsTensorIdRegular(const TensorId& tensor_id) { in IsTensorIdRegular() 233 TensorId fanin_id = ParseTensorName(input); in AddNode() 332 const TensorId& fanin) { in AddOrUpdateRegularFanin() 339 const TensorId& fanin) { in AddOrUpdateRegularFanin() 478 std::vector<std::vector<TensorId>> fanins; in MutableGraphView() 503 std::vector<std::vector<TensorId>>* fanins) { in CheckFaninsInternal() 510 std::vector<TensorId> node_fanins; in CheckFaninsInternal() 513 TensorId fanin_id = ParseTensorName(input); in CheckFaninsInternal() 541 std::vector<std::vector<TensorId>>* fanins) { in AddFaninsInternal() [all …]
|
D | grappler_test.cc | 65 const TensorId want_tensor = ParseTensorName(want_node.input(j)); in CompareGraphNodes() 66 const TensorId got_tensor = ParseTensorName(got_node.input(j)); in CompareGraphNodes()
|
/external/tensorflow/tensorflow/core/grappler/optimizers/ |
D | dependency_optimizer.cc | 171 const TensorId consumer_input = ParseTensorName(consumer->input(j)); in NumEdgesIfBypassed() 263 const TensorId input_tensor = ParseTensorName(fanout->input(i)); in OptimizeNode() 411 const TensorId old_input = ParseTensorName(consumer->input(j)); in OptimizeNode()
|