Searched refs:BilinearWeights (Results 1 – 1 of 1) sorted by relevance
52 XlaOp BilinearWeights(XlaOpKernelContext* ctx, XlaOp ratio, in BilinearWeights() function254 auto weights = BilinearWeights(ctx, ratio, warp_shape, warp_type); in CalculateGradData()530 auto weights = BilinearWeights(ctx, ratio, warp_shape, data_type); in Compile()