Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/core/graph/
Dtensor_id.h33 struct TensorId : public std::pair<StringPiece, int> { struct
41 TensorId() : Base() {} in TensorId() function
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()
Dgraph_constructor_test.cc1180 opts.input_map[TensorId("new_input", 0)] = TensorId("input", 1); in TEST_F()
1181 opts.input_map[TensorId("new_input", 1)] = TensorId("input", 0); in TEST_F()
1222 opts.input_map[TensorId("input", 0)] = TensorId("input", 0); in TEST_F()
1223 opts.input_map[TensorId("input", 1)] = TensorId("input", 0); in TEST_F()
1284 opts.input_map[TensorId("W2", kControlSlot)] = TensorId("W1", kControlSlot); in TEST_F()
1285 opts.input_map[TensorId("W3", kControlSlot)] = TensorId("W1", kControlSlot); in TEST_F()
1321 opts.input_map[TensorId("W1", kControlSlot)] = TensorId("W1", kControlSlot); in TEST_F()
1348 opts.input_map[TensorId("W2", Graph::kControlSlot)] = TensorId("W1", 0); in TEST_F()
1360 opts.input_map[TensorId("W2", 0)] = TensorId("W1", Graph::kControlSlot); in TEST_F()
1380 opts.input_map[TensorId("input2", 0)] = TensorId("input1", 3); in TEST_F()
[all …]
Dgraph_constructor.cc100 std::map<TensorId, TensorId> input_map;
103 std::vector<TensorId> return_tensors;
263 std::set<TensorId> used_input_map_keys_;
346 bool NodeNameInValues(const std::map<TensorId, TensorId>& input_map, in NodeNameInValues() argument
414 TensorId src = mapping.first; in ValidateInputMapAndControlDependencies()
415 TensorId dst = mapping.second; in ValidateInputMapAndControlDependencies()
518 TensorId id(ParseTensorName(input_name)); in InitFromEdges()
531 TensorId id(ParseTensorName(input_name)); in InitFromEdges()
717 std::set<TensorId> control_inputs; in RemapNodeDefInputs()
725 TensorId new_input = iter->second; in RemapNodeDefInputs()
[all …]
Dtensor_id_test.cc82 TensorId id; in BM_ParseTensorName()
95 TensorId tensor_id = ParseTensorName(input); in TEST()
Dsubgraph.cc65 TensorId id(ParseTensorName(t)); in FeedInputs()
133 TensorId id(ParseTensorName(t)); in FetchOutputs()
180 TensorId id = ParseTensorName(node_or_tensor_name); in AddNodeToTargets()
/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()
122 TensorId tensor_id = ParseTensorName(control_fanout.node->input(i)); in SwapControlledFanoutInputs()
266 Status CheckFaninIsRegular(const TensorId& fanin, ErrorHandler handler) { in CheckFaninIsRegular()
274 Status CheckFaninIsValid(const TensorId& fanin, ErrorHandler handler) { in CheckFaninIsValid()
283 const TensorId& fanin, ErrorHandler handler) { in CheckAddingFaninToSelf()
292 const TensorId& fanin, ErrorHandler handler) { in CheckRemovingFaninFromSelf()
348 TensorId tensor_id = ParseTensorName(node->input(pos)); in AddAndDedupFanouts()
[all …]
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.cc160 string TensorIdToString(const TensorId& tensor_id) { in TensorIdToString()
167 TensorId tensor1 = ParseTensorName(name1); in IsSameInput()
168 TensorId tensor2 = ParseTensorName(name2); in IsSameInput()
176 bool IsControlInput(const TensorId& tensor_id) { return tensor_id.index() < 0; } in IsControlInput()
284 const TensorId tensor = ParseTensorName(node_as_input); in NumNonControlOutputs()
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 …]
Dutils.h105 string TensorIdToString(const TensorId& tensor_id);
112 bool IsControlInput(const TensorId& tensor_id);
/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.cc272 explicit SymbolPredicate(int64 id, TensorId tensor_id, bool must_be_true) in SymbolPredicate()
290 TensorId tensor_id() const { return tensor_id_; } in tensor_id()
294 TensorId tensor_id_;
370 TensorId tensor_id(node->name(), output_idx); in MakeSymbolPredicate()
703 absl::flat_hash_map<TensorId, string, TensorId::Hasher> PredicateMapAsString()
718 predicate_map_.insert({TensorId(n->name(), output_idx), pred}); in SetPredicate()
747 absl::flat_hash_map<TensorId, Predicate*, TensorId::Hasher> predicate_map_;
753 TensorId InputEdgeToTensorId(const Edge* e) { in InputEdgeToTensorId()
754 return TensorId(e->src()->name(), e->src_output()); in InputEdgeToTensorId()
930 auto it = predicate_map_.find(TensorId(n->name(), 0)); in HandleMerge()
[all …]
Ddeadness_analysis_internal.h27 using PredicateMapTy = absl::flat_hash_map<TensorId, string, TensorId::Hasher>;
/external/tensorflow/tensorflow/contrib/android/java/org/tensorflow/contrib/android/
DTensorFlowInferenceInterface.java208 TensorId tid = TensorId.parse(o); in run()
576 TensorId tid = TensorId.parse(inputName); in addFeed()
582 private static class TensorId { class in TensorFlowInferenceInterface
589 public static TensorId parse(String name) { in parse()
590 TensorId tid = new TensorId(); in parse()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dtf2xla.proto12 // TensorId identifies a tensor in a TensorFlow graph, by specifying the output
16 message TensorId { message
24 TensorId id = 1;
38 TensorId id = 1;
Dtf2xla_util.h50 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()
524 const TensorId tid = ParseTensorName(node_name); in GetTensorShapeType()
569 const TensorId tid = ParseTensorName(name); in EmplaceTensorShapeType()
693 const TensorId tid = ParseTensorName(input); in BuildRemoteFusedGraphExecuteOpNode()
[all …]
/external/tensorflow/tensorflow/core/grappler/utils/
Dgrappler_test.cc62 const TensorId want_tensor = ParseTensorName(want_node.input(j)); in CompareGraphNodes()
63 const TensorId got_tensor = ParseTensorName(got_node.input(j)); in CompareGraphNodes()
/external/tensorflow/tensorflow/core/common_runtime/
Dsession_state.cc72 TensorId id(ParseTensorName(name)); in SaveTensors()
Dgraph_execution_state.cc324 const TensorId tensor;
507 TensorId from_id(ParseTensorName(tensor_connection.from_tensor())); in PruneGraph()
635 TensorId id = ParseTensorName(feed); in OptimizeGraph()
643 TensorId id = ParseTensorName(tensor_connection.to_tensor()); in OptimizeGraph()
Ddirect_session.cc1132 std::unordered_set<TensorId, TensorId::Hasher> pending_feeds; in CheckFetch()
1138 TensorId id(ParseTensorName(input.first)); in CheckFetch()
1147 TensorId id(ParseTensorName(it.first)); in CheckFetch()
1154 TensorId id(ParseTensorName(fetch)); in CheckFetch()
1216 TensorId id(ParseTensorName(input)); in CreateExecutors()
1220 TensorId id(ParseTensorName(output)); in CreateExecutors()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Ddependency_optimizer.cc149 const TensorId consumer_input = ParseTensorName(consumer->input(j)); in NumEdgesIfBypassed()
241 const TensorId input_tensor = ParseTensorName(fanout->input(i)); in OptimizeNode()
387 const TensorId old_input = ParseTensorName(consumer->input(j)); in OptimizeNode()
/external/tensorflow/tensorflow/core/grappler/costs/
Dutils.cc96 TensorId input_tensor_id = ParseTensorName(input_name); in ExtractExtraProperties()
169 TensorId input_tensor_id = ParseTensorName(input_name); in FindInputFeatures()

12