Searched refs:BroadcastSpatialDimensions (Results 1 – 1 of 1) sorted by relevance
209 xla::XlaOp BroadcastSpatialDimensions(xla::XlaBuilder* builder, in BroadcastSpatialDimensions() function373 return BroadcastSpatialDimensions( in ResizeUsingDilationAndConvolution()407 kernel = BroadcastSpatialDimensions( in ResizeUsingDilationAndConvolutionGradOp()428 kernel0 = BroadcastSpatialDimensions(builder, kernel0, in ResizeUsingDilationAndConvolutionGradOp()433 kernel1 = BroadcastSpatialDimensions(builder, kernel0, in ResizeUsingDilationAndConvolutionGradOp()