Searched refs:num_bcast_dims (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | einsum_op_impl.h | 184 static void InsertBroadcastLabels(int num_bcast_dims, int num_named_labels, in InsertBroadcastLabels()
188 labels->insert(labels->begin() + ellipsis_axis, num_bcast_dims, 0); in InsertBroadcastLabels()
190 labels->begin() + ellipsis_axis + num_bcast_dims, in InsertBroadcastLabels()
194 label_counts->resize(num_named_labels + num_bcast_dims, 1); in InsertBroadcastLabels()
260 const int num_bcast_dims = inputs[i].dims() - labels->size() + 1; in ProcessDimensions() local
269 label_idx + (ellipsis_axis == -1 ? 0 : num_bcast_dims - 1); in ProcessDimensions()
276 InsertBroadcastLabels(num_bcast_dims, num_named_labels, ellipsis_axis, in ProcessDimensions()
278 max_bcast_dims = std::max(max_bcast_dims, num_bcast_dims); in ProcessDimensions()
|