Searched refs:num_named_labels (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | einsum_op_impl.h | 64 static void InsertBroadcastLabels(int num_bcast_dims, int num_named_labels, in InsertBroadcastLabels()
71 num_named_labels); in InsertBroadcastLabels()
74 label_counts->resize(num_named_labels + num_bcast_dims, 1); in InsertBroadcastLabels()
114 const int num_named_labels = label_types->size(); in ProcessDimensions() local
115 label_to_dim_sizes->resize(num_named_labels); in ProcessDimensions()
156 InsertBroadcastLabels(num_bcast_dims, num_named_labels, ellipsis_axis, in ProcessDimensions()
170 InsertBroadcastLabels(max_bcast_dims, num_named_labels, ellipsis_axis, in ProcessDimensions()
179 label_types->resize(num_named_labels + max_bcast_dims, in ProcessDimensions()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 189 const int num_named_labels = in EinsumShape() local
192 c->WithRankAtLeast(input_shape, num_named_labels, &input_shape), in EinsumShape()
195 const int num_named_labels = static_cast<int>(input_labels[i].size()); in EinsumShape() local
197 c->WithRank(input_shape, num_named_labels, &input_shape), " for ", in EinsumShape()
|