Searched refs:LocalShapedBufferTuple (Results 1 – 3 of 3) sorted by relevance
70 class LocalShapedBufferTuple; variable92 StatusOr<LocalShapedBufferTuple*> DestructureTuple();103 class LocalShapedBufferTuple {107 explicit LocalShapedBufferTuple(std::vector<LocalShapedBuffer*> elements);109 ~LocalShapedBufferTuple();143 StatusOr<LocalShapedBufferTuple*> ExecutePerReplica(
137 LocalShapedBufferTuple::LocalShapedBufferTuple( in LocalShapedBufferTuple() function in xla::swig::LocalShapedBufferTuple145 LocalShapedBufferTuple::~LocalShapedBufferTuple() { in ~LocalShapedBufferTuple()153 StatusOr<LocalShapedBuffer*> LocalShapedBufferTuple::Release(int i) { in Release()163 int64 LocalShapedBufferTuple::size() const { return elements_.size(); } in size()165 StatusOr<LocalShapedBufferTuple*> LocalShapedBuffer::DestructureTuple() { in DestructureTuple()207 return new LocalShapedBufferTuple(std::move(results)); in DestructureTuple()264 StatusOr<LocalShapedBufferTuple*> LocalExecutable::ExecutePerReplica( in ExecutePerReplica()333 return new LocalShapedBufferTuple(std::move(wrapped_results)); in ExecutePerReplica()
114 %typemap(out) StatusOr<xla::swig::LocalShapedBufferTuple*> {119 $typemap(out, xla::swig::LocalShapedBufferTuple*)269 %unignore xla::swig::LocalShapedBufferTuple;270 %unignore xla::swig::LocalShapedBufferTuple::Release;271 %unignore xla::swig::LocalShapedBufferTuple::size;