Searched refs:RecordInShapeLayoutCache (Results 1 – 1 of 1) sorted by relevance
368 void RecordInShapeLayoutCache(const TensorWithLayout& tensor);523 void DTensorDevice::RecordInShapeLayoutCache(const TensorWithLayout& tensor) { in RecordInShapeLayoutCache() function in tensorflow::dtensor::DTensorDevice741 RecordInShapeLayoutCache(*wrapper); in CopyToMesh()908 RecordInShapeLayoutCache(*packed_tensor); in Pack()1054 RecordInShapeLayoutCache(*packed_tensor); in SparsePack()1770 RecordInShapeLayoutCache(*output_with_layout[i]); in ExecuteRegularOperation()