Searched refs:label_idx (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | einsum_op_impl.h | 245 for (int label_idx = 0; label_idx < labels->size(); ++label_idx) { in ProcessDimensions() local
246 const int label = (*labels)[label_idx]; in ProcessDimensions()
247 TF_RETURN_IF_ERROR(RecordLabelToDimension(label, label_idx, inputs[i], in ProcessDimensions()
261 for (int label_idx = 0; label_idx < labels->size(); ++label_idx) { in ProcessDimensions() local
262 const int label = (*labels)[label_idx]; in ProcessDimensions()
264 ellipsis_axis = label_idx; in ProcessDimensions()
269 label_idx + (ellipsis_axis == -1 ? 0 : num_bcast_dims - 1); in ProcessDimensions()
498 for (int label_idx = 0; label_idx < labels->size(); ++label_idx) { in ReduceOperand() local
499 const int label = labels->at(label_idx); in ReduceOperand()
500 int64 dim = input_deduped.dim_size(label_idx); in ReduceOperand()
|
| /external/mesa3d/src/freedreno/afuc/ |
| D | disasm.c | 220 static int label_idx(uint32_t offset, bool create) in label_idx() function
255 lidx = label_idx(offset, false); in label_name()
332 label_idx(i + instr->br.ioff, true); in disasm()
336 label_idx(instr->call.uoff, true); in disasm()
343 label_idx(i + 3, true); in disasm()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 206 for (int label_idx = 0, end = input_labels[i].size(); label_idx < end; in EinsumShape() local
207 ++label_idx) { in EinsumShape()
208 const char label = input_labels[i][label_idx]; in EinsumShape()
212 const int64 axis_before_ellipsis = label_idx; in EinsumShape()
215 ? label_idx + c->Rank(input_shape) - input_labels[i].size() in EinsumShape()
229 label_idx += 2; // Skip the rest of the ellipsis. in EinsumShape()
286 for (int label_idx = 0, end = output_labels.size(); label_idx < end; in EinsumShape() local
287 ++label_idx) { in EinsumShape()
288 const char label = output_labels[label_idx]; in EinsumShape()
294 label_idx += 2; // Skip the rest of the ellipsis. in EinsumShape()
|
| /external/libaom/libaom/av1/encoder/ |
| D | partition_strategy.h | 37 int bsize, int label_idx,
|