Searched refs:TensorDesc (Results 1 – 1 of 1) sorted by relevance
47 typedef std::pair<Node*, int> TensorDesc; typedef51 void ReplaceEdgeSources(const TensorDesc& old_src, const TensorDesc& new_src, in ReplaceEdgeSources()79 const TensorDesc& output) { in AddMapDefunOutput()183 Status StackTensor(WrappedTensor* unstacked, TensorDesc* result);200 TensorDesc&& tensor, absl::flat_hash_set<const Edge*>* visited);208 std::map<TensorDesc, WrappedTensor> conversion_map_;304 TensorDesc output({ret_edge->src(), ret_edge->src_output()}); in ConvertOutput()305 TensorDesc converted_output; in ConvertOutput()417 TensorDesc* result) { in StackTensor()539 TensorDesc&& tensor, absl::flat_hash_set<const Edge*>* visited) { in AddUnstackedTensorMappingsHelper()