Searched refs:SafeTensorId (Results 1 – 9 of 9) sorted by relevance
| /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() function
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()
|
| D | graph_constructor.h | 91 std::map<SafeTensorId, SafeTensorId> input_map;
115 std::vector<SafeTensorId> return_tensors;
161 std::vector<SafeTensorId> missing_unused_input_map_keys;
|
| D | graph_constructor.cc | 129 std::vector<SafeTensorId>* missing_unused_input_map_keys) { in Construct()
150 std::vector<SafeTensorId>* missing_unused_input_map_keys) in GraphConstructor()
259 std::vector<SafeTensorId>* missing_unused_input_map_keys_;
|
| D | graph_constructor_test.cc | 1512 SafeTensorId("new_input", 2)); in TEST_F()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | function_optimizer.cc | 301 const absl::flat_hash_map<SafeTensorId, SafeTensorId, SafeTensorId::Hasher>&
355 void AddTensorMapping(const SafeTensorId& from, const SafeTensorId& to) { in AddTensorMapping()
373 SafeTensorId from_tensor(func_node, from_idx); in AddTensorMapping()
374 SafeTensorId to_tensor(func_node, to_idx); in AddTensorMapping()
456 absl::flat_hash_map<SafeTensorId, SafeTensorId, SafeTensorId::Hasher>
1627 std::vector<SafeTensorId> inputs; in InlineIndirectFunctionCall()
1629 SafeTensorId tensor_id = ParseTensorName(input); in InlineIndirectFunctionCall()
1670 SafeTensorId tensor_id = ParseTensorName(output_tensor); in InlineIndirectFunctionCall()
1857 SafeTensorId tensor = ParseTensorName(input); in InlineIndirectFunctionCall()
1903 const SafeTensorId from_tensor(func_node.name(), output_idx++); in InlineIndirectFunctionCall()
[all …]
|
| D | graph_optimizer_stage.cc | 52 SafeTensorId tensor_id = ParseTensorName(tensor); in GetTensorProperties()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | deadness_analysis.cc | 479 using SignatureForSymbol = std::pair<SafeTensorId, bool>;
493 return Hash64Combine(SafeTensorId::Hasher()(signature.first), in operator ()()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties.cc | 2045 SafeTensorId tensor_id = ParseTensorName(feed.first); in InferStatically()
|