| /external/tensorflow/tensorflow/core/graph/ |
| D | tensor_id.h | 28 struct SafeTensorId;
42 TensorId(const SafeTensorId& id);
67 struct SafeTensorId : public std::pair<string, int> { struct
72 SafeTensorId() : Base() {} in SafeTensorId() argument
73 SafeTensorId(const string& str, int idx) : Base(str, idx) {} in SafeTensorId() argument
74 SafeTensorId(const TensorId& id);
|
| D | tensor_id.cc | 25 TensorId::TensorId(const SafeTensorId& id) : TensorId(id.first, id.second) {} in TensorId()
27 SafeTensorId::SafeTensorId(const TensorId& id) in SafeTensorId() function in tensorflow::SafeTensorId
28 : SafeTensorId(string(id.first), id.second) {} in SafeTensorId()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | graph_constructor.h | 103 std::map<SafeTensorId, SafeTensorId> input_map;
127 std::vector<SafeTensorId> return_tensors;
173 std::vector<SafeTensorId> missing_unused_input_map_keys;
|
| D | graph_constructor.cc | 164 std::vector<SafeTensorId>* missing_unused_input_map_keys);
170 std::vector<SafeTensorId>* missing_unused_input_map_keys);
176 std::vector<SafeTensorId>* missing_unused_input_map_keys) in GraphConstructor()
312 std::vector<SafeTensorId>* missing_unused_input_map_keys_;
405 std::vector<SafeTensorId>* missing_unused_input_map_keys) in NodeDefCopyingGraphConstructor()
432 std::vector<SafeTensorId>* missing_unused_input_map_keys) in NodeDefMovingGraphConstructor()
491 std::vector<SafeTensorId>* missing_unused_input_map_keys) { in Construct()
512 std::vector<SafeTensorId>* missing_unused_input_map_keys) { in Construct()
|
| D | graph_execution_state.cc | 688 std::vector<SafeTensorId> feeds; in OptimizeGraph()
705 for (const SafeTensorId& feed : feeds) { in OptimizeGraph()
775 SafeTensorId tensor_id = ParseTensorName(feed.first); in OptimizeGraph()
782 SafeTensorId tensor_id = ParseTensorName(fetch); in OptimizeGraph()
|
| D | graph_constructor_test.cc | 1516 SafeTensorId("new_input", 2)); in TEST_F()
|
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | graph_view_internal.h | 353 inline SafeTensorId EmptyTensorId() { in EmptyTensorId()
354 return SafeTensorId("", internal::kMissingSlot); in EmptyTensorId()
378 std::vector<SafeTensorId> regular_inputs_to_add;
384 std::map<int, SafeTensorId> regular_inputs_to_update;
515 SafeTensorId(fanin)); in AddOrUpdateRegularFanin()
664 std::vector<SafeTensorId>().swap(diff->regular_inputs_to_add); in Reset()
666 std::map<int, SafeTensorId>().swap(diff->regular_inputs_to_update); in Reset()
788 std::vector<SafeTensorId> regular_fanins;
|
| D | graph_view.h | 498 const SafeTensorId& fanin_id);
502 const SafeTensorId& fanin_id);
|
| D | graph_view.cc | 228 std::vector<SafeTensorId> regular_fanins; in AddNode()
1061 MutableNodeView* node_view, const SafeTensorId& fanin_id) { in AddRegularFaninInternal()
1086 MutableNodeView* node_view, const int i, const SafeTensorId& fanin_id) { in UpdateRegularFaninInternal()
1233 for (const SafeTensorId& fanin : diff.regular_inputs_to_add) { in ApplyNodeUpdates()
1282 for (const SafeTensorId& fanin : new_node->regular_fanins) { in SetNewNodesFanins()
|
| D | functions.cc | 477 const SafeTensorId tensor = ParseTensorName(graph_def_input); in AsFunctionDefInput()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | function_optimizer.cc | 279 const absl::flat_hash_map<SafeTensorId, SafeTensorId, SafeTensorId::Hasher>&
317 void AddTensorMapping(const SafeTensorId& from, const SafeTensorId& to) { in AddTensorMapping()
335 SafeTensorId from_tensor(func_node, from_idx); in AddTensorMapping()
336 SafeTensorId to_tensor(func_node, to_idx); in AddTensorMapping()
379 absl::flat_hash_map<SafeTensorId, SafeTensorId, SafeTensorId::Hasher>
|
| D | graph_optimizer_stage.cc | 52 SafeTensorId tensor_id = ParseTensorName(tensor); in GetTensorProperties()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | deadness_analysis.cc | 562 using SignatureForSymbol = std::pair<SafeTensorId, bool>;
563 using SignatureForIntSymbol = std::pair<SafeTensorId, std::optional<int32>>;
577 return Hash64Combine(SafeTensorId::Hasher()(signature.first), in operator ()()
585 SafeTensorId::Hasher()(signature.first), in operator ()()
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | utils.h | 298 string SafeTensorIdToString(const SafeTensorId& tensor_id);
|
| D | utils.cc | 81 string SafeTensorIdToString(const SafeTensorId& tensor_id) { in SafeTensorIdToString()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties.cc | 2524 SafeTensorId tensor_id = ParseTensorName(feed.first); in InferStatically()
|