Searched refs:dest_idx (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/contrib/distributions/python/ops/ |
| D | distribution_util.py | 497 def move_dimension(x, source_idx, dest_idx): argument
535 if dest_idx < 0:
536 dest_idx = ndims + dest_idx
543 math_ops.range(0, dest_idx, dtype=dtype),
545 math_ops.range(dest_idx, source_idx, dtype=dtype),
552 math_ops.range(source_idx+1, dest_idx+1, dtype=dtype),
554 math_ops.range(dest_idx+1, ndims, dtype=dtype)], axis=0))
558 x, perm=smart_cond.smart_cond(source_idx < dest_idx,
564 return smart_cond.smart_cond(math_ops.equal(source_idx, dest_idx),
|
| /external/mesa3d/src/compiler/nir/ |
| D | nir_from_ssa.c | 641 int dest_idx = -1; in resolve_parallel_copy() local
650 dest_idx = i; in resolve_parallel_copy()
653 if (dest_idx < 0) { in resolve_parallel_copy()
654 dest_idx = num_vals++; in resolve_parallel_copy()
655 values[dest_idx] = dest_src; in resolve_parallel_copy()
659 pred[dest_idx] = src_idx; in resolve_parallel_copy()
661 to_do[++to_do_idx] = dest_idx; in resolve_parallel_copy()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | resource_variable_ops.cc | 603 for (int64 batch_idx = 0, dest_idx = 0; batch_idx < batch_size; in AddBatchOffsets() local
606 indices_flat(dest_idx++) += batch_offset * batch_idx; in AddBatchOffsets()
|
| /external/mesa3d/src/compiler/spirv/ |
| D | spirv_to_nir.c | 2802 unsigned dest_idx = 0; in vtn_vector_construct() local
2805 vtn_assert(dest_idx + src->num_components <= num_components); in vtn_vector_construct()
2807 vec->src[dest_idx].src = nir_src_for_ssa(src); in vtn_vector_construct()
2808 vec->src[dest_idx].swizzle[0] = j; in vtn_vector_construct()
2809 dest_idx++; in vtn_vector_construct()
2818 vtn_assert(dest_idx == num_components); in vtn_vector_construct()
|