Searched defs:warp_shape (Results 1 – 2 of 2) sorted by relevance
53 const TensorShape warp_shape, in BilinearWeights()118 const TensorShape& warp_shape) { in ConcatenateIota()196 xla::PrimitiveType warp_type, TensorShape warp_shape, in BoundSamples()250 xla::PrimitiveType warp_type, TensorShape warp_shape, in CalculateGradData()361 TensorShape warp_shape, int64 data_channels, in CalculateGradWarp()495 TensorShape warp_shape = ctx->InputShape("warp"); in Compile() local616 TensorShape warp_shape = ctx->InputShape("warp"); in Compile() local
142 const ::tensorflow::TensorShape& warp_shape = warp.shape(); in Compute() local352 const ::tensorflow::TensorShape& warp_shape = warp.shape(); in Compute() local