Searched defs:broadcast_dim (Results 1 – 7 of 7) sorted by relevance
31 inline int broadcast_dim(int lhs_dim, int rhs_dim) { in broadcast_dim() function
109 int broadcast_dim = lhs_dim; in ResizeOutputTensor() local
893 for (int64 broadcast_dim : broadcast_dims) { in ComputeArrayForElementwiseBinaryOp() local
1374 HloInstruction* broadcast_dim = comp->AddInstruction( in RewriteDynamicUpdateSlice() local
239 int64 broadcast_dim = hlo->dimensions(dimension); in HandleBroadcast() local
673 for (const auto& broadcast_dim : llvm::enumerate(broadcast_dims)) { in getIndexingMaps() local
4008 int64_t broadcast_dim; in matchAndRewrite() local