Searched defs:zero_r2 (Results 1 – 2 of 2) sorted by relevance
88 xla::XlaOp zero_r2 = xla::Broadcast(zero_r0, predictions_shape.dim_sizes()); in Compile() local
52 xla::XlaOp zero_r2 = xla::Broadcast(zero_r0, {height, width}); in CreateKthOrderStatisticComputation() local264 xla::XlaOp zero_r2 = xla::Broadcast(zero_r0, {height, width}); in CreateMakeUnique() local