Searched refs:GetAllocationSliceForMlir (Results 1 – 4 of 4) sorted by relevance
623 StatusOr<BufferAllocation::Slice> IrEmitterUnnested::GetAllocationSliceForMlir( in GetAllocationSliceForMlir() function in xla::gpu::IrEmitterUnnested625 return xla::gpu::GetAllocationSliceForMlir( in GetAllocationSliceForMlir()1123 TF_ASSIGN_OR_RETURN(auto slice, GetAllocationSliceForMlir(operand)); in EmitConvolutionThunkFromMlir()1130 GetAllocationSliceForMlir(conv_result)); in EmitConvolutionThunkFromMlir()1132 GetAllocationSliceForMlir(scratch_result)); in EmitConvolutionThunkFromMlir()1261 TF_ASSIGN_OR_RETURN(lhs, GetAllocationSliceForMlir(gemm.lhs())); in EmitGemmThunkFromMlir()1262 TF_ASSIGN_OR_RETURN(rhs, GetAllocationSliceForMlir(gemm.rhs())); in EmitGemmThunkFromMlir()1263 TF_ASSIGN_OR_RETURN(output, GetAllocationSliceForMlir(gemm.output())); in EmitGemmThunkFromMlir()1268 TF_ASSIGN_OR_RETURN(lhs, GetAllocationSliceForMlir(gemm_bias.lhs())); in EmitGemmThunkFromMlir()1269 TF_ASSIGN_OR_RETURN(rhs, GetAllocationSliceForMlir(gemm_bias.rhs())); in EmitGemmThunkFromMlir()[all …]
263 StatusOr<BufferAllocation::Slice> GetAllocationSliceForMlir(
740 StatusOr<BufferAllocation::Slice> GetAllocationSliceForMlir( in GetAllocationSliceForMlir() function810 auto maybe_lhs = GetAllocationSliceForMlir(parameter.memref(), allocations); in CanEmitFusedDynamicUpdateSliceInPlaceForGpu()811 auto maybe_rhs = GetAllocationSliceForMlir(output_buffers[0], allocations); in CanEmitFusedDynamicUpdateSliceInPlaceForGpu()
352 StatusOr<BufferAllocation::Slice> GetAllocationSliceForMlir(mlir::Value v);