Searched refs:SubBuffer (Results 1 – 3 of 3) sorted by relevance
824 class SubBuffer : public TensorBuffer { class827 SubBuffer(TensorBuffer* buf, int64 delta, int64 n) in SubBuffer() function in tensorflow::SubBuffer854 ~SubBuffer() override { root_->Unref(); } in ~SubBuffer()856 TF_DISALLOW_COPY_AND_ASSIGN(SubBuffer);880 CASES(dt, ret.buf_ = new SubBuffer<T>(buf_, delta, num_elems)); in Slice()902 CASES(dt, ret.buf_ = new SubBuffer<T>(buf_, delta, num_elems)); in SubSlice()
280 cl_mem SubBuffer = in getDeviceMemorySpanHandle() local286 return SubBuffer; in getDeviceMemorySpanHandle()
768 TEST(RawApiTest, SubBuffer) { in TEST() argument