Home
last modified time | relevance | path

Searched refs:TensorId (Results 1 – 25 of 53) sorted by relevance

123

/external/tensorflow/tensorflow/core/graph/
Dtensor_id.h33 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()
Dtensor_id.cc25 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()
Dtensor_id_test.cc82 TensorId id; in BM_ParseTensorName()
95 TensorId tensor_id = ParseTensorName(input); in TEST()
Dsubgraph.cc64 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/
Dgraph_constructor_test.cc1184 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 …]
Dgraph_constructor.cc126 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 …]
Dsession_state.cc76 TensorId id(ParseTensorName(name)); in SaveTensors()
/external/tensorflow/tensorflow/core/grappler/
Dmutable_graph_view.h141 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);
Dmutable_graph_view.cc43 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 …]
Dutils.cc91 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()
Dgraph_view.h159 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()
Dutils.h58 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);
Dmutable_graph_view_test.cc40 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/
Dfold_constants_lib.cc53 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/
Ddeadness_analysis.cc283 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 …]
Ddeadness_analysis_internal.h27 using PredicateMapTy = absl::flat_hash_map<TensorId, string, TensorId::Hasher>;
Dintroduce_floating_point_jitter_pass.cc37 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/
DTensorFlowInferenceInterface.java208 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/
Dtf2xla.proto13 // TensorId identifies a tensor in a TensorFlow graph, by specifying the output
17 message TensorId { message
25 TensorId id = 1;
39 TensorId id = 1;
Dtf2xla_util.h51 string TensorIdToString(const tf2xla::TensorId& id);
/external/tensorflow/tensorflow/core/kernels/
Dremote_fused_graph_execute_utils.cc49 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/
Dgraph_view.h298 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);
Dgraph_view.cc127 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 …]
Dgrappler_test.cc65 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/
Ddependency_optimizer.cc171 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()

123