Searched defs:broadcast_dim (Results 1 – 9 of 9) sorted by relevance
84 int32_t broadcast_dim = 0; in get_broadcast_dim() local128 int32_t broadcast_dim = get_broadcast_dim(lhs, rhs); in select_broadcast_optimized_path() local174 const int32_t broadcast_dim) { in get_normalized_tensor_size()
155 int32_t broadcast_dim = internal::get_broadcast_dim(*lhs, *rhs); in handle_broadcast_mul() local
732 def broadcast_dim(tensor_input1, tensor_input2, res1, res2, index, padding=False): function
31 inline int broadcast_dim(int lhs_dim, int rhs_dim) { in broadcast_dim() function
109 int broadcast_dim = lhs_dim; in ResizeOutputTensor() local
897 for (int64_t broadcast_dim : broadcast_dims) { in ComputeArrayForElementwiseBinaryOp() local
1632 HloInstruction* broadcast_dim = dus->AddInstruction( in RewriteDynamicUpdateSlice() local
270 int64_t broadcast_dim = hlo->dimensions(dimension); in HandleBroadcast() local
4665 int64_t broadcast_dim; in matchAndRewrite() local