Searched refs:scratch_slice (Results 1 – 4 of 4) sorted by relevance
35 BufferAllocation::Slice result_slice, BufferAllocation::Slice scratch_slice, in ConvolutionThunk() argument41 scratch_buffer_(scratch_slice), in ConvolutionThunk()
49 BufferAllocation::Slice scratch_slice,
476 auto scratch_slice = assn.GetUniqueSlice(custom_call, {1}).ValueOrDie(); in HandleCustomCall() local480 conv_result_slice, scratch_slice, tuple_result_slice)); in HandleCustomCall()
121 const InputBuffer scratch_slice(out_flat.data() + begin, len); in Compute() local126 fns_[i](scratch_slice, &out_slice); in Compute()