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_t broadcast_dim : broadcast_dims) { in ComputeArrayForElementwiseBinaryOp() local
1566 HloInstruction* broadcast_dim = comp->AddInstruction( in RewriteDynamicUpdateSlice() local
241 int64_t broadcast_dim = hlo->dimensions(dimension); in HandleBroadcast() local
865 for (const auto& broadcast_dim : in matchAndRewrite() local1033 for (const auto& broadcast_dim : llvm::enumerate(broadcast_dims)) { in getIndexingMaps() local
4815 int64_t broadcast_dim; in matchAndRewrite() local