Searched refs:make_kernel_func (Results 1 – 1 of 1) sorted by relevance
179 auto make_kernel_func = in MakeGeneralResizeKernel() local185 xla::BroadcastInDim(make_kernel_func(builder, type, kernel_size[1]), in MakeGeneralResizeKernel()189 make_kernel_func(builder, type, kernel_size[0]), in MakeGeneralResizeKernel()198 auto make_kernel_func = in MakeGeneralResizeKernelInDim() local204 return xla::BroadcastInDim(make_kernel_func(builder, type, kernel_size[dim]), in MakeGeneralResizeKernelInDim()