Searched defs:scratch_slice (Results 1 – 3 of 3) sorted by relevance
35 BufferAllocation::Slice result_slice, BufferAllocation::Slice scratch_slice, in ConvolutionThunk()
476 auto scratch_slice = assn.GetUniqueSlice(custom_call, {1}).ValueOrDie(); in HandleCustomCall() local
121 const InputBuffer scratch_slice(out_flat.data() + begin, len); in Compute() local