Searched refs:scratch_slice (Results 1 – 4 of 4) sorted by relevance
36 BufferAllocation::Slice result_slice, BufferAllocation::Slice scratch_slice) in ConvolutionThunk() argument40 scratch_buffer_(scratch_slice), in ConvolutionThunk()
48 BufferAllocation::Slice scratch_slice);
1131 TF_ASSIGN_OR_RETURN(auto scratch_slice, in EmitConvolutionThunkFromMlir()1219 conv_result_slice, scratch_slice)); in EmitConvolutionThunkFromMlir()
121 const InputBuffer scratch_slice(out_flat.data() + begin, len); in Compute() local126 fns_[i](scratch_slice, &out_slice); in Compute()