Home
last modified time | relevance | path

Searched refs:row_idx (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/contrib/boosted_trees/lib/utils/
Dexamples_iterable.h123 for (int64 row_idx = row_range.start; row_idx < row_range.end; variable
124 ++row_idx) {
126 const int32 feature_id = indices(row_idx, 1);
127 DCHECK_EQ(example_idx_, indices(row_idx, 0));
133 row_idx));
149 for (int64 row_idx = row_range.start; row_idx < row_range.end; variable
150 ++row_idx) {
152 iter_->sparse_int_column_values_[sparse_int_idx](row_idx));
Dsparse_column_iterable.cc32 IndicesRowIterator(SparseColumnIterable* iter, int row_idx) in IndicesRowIterator() argument
33 : iter_(iter), row_idx_(row_idx) {} in IndicesRowIterator()
108 Eigen::Index row_idx() const { return row_idx_; } in row_idx() function in tensorflow::boosted_trees::utils::__anon7430c01b0111::IndicesRowIterator
120 .row_idx(); in Iterator()
/external/skia/src/compute/hs/cl/intel/
Dhs_cl_macros.h106 #define HS_SLAB_GLOBAL_LOAD(extent,row_idx) \ argument
107 extent[gmem_idx + HS_SLAB_THREADS * row_idx]
109 #define HS_SLAB_GLOBAL_STORE(row_idx,reg) \ argument
110 vout[gmem_idx + HS_SLAB_THREADS * row_idx] = reg
/external/skqp/src/compute/hs/cl/intel/
Dhs_cl_macros.h106 #define HS_SLAB_GLOBAL_LOAD(extent,row_idx) \ argument
107 extent[gmem_idx + HS_SLAB_THREADS * row_idx]
109 #define HS_SLAB_GLOBAL_STORE(row_idx,reg) \ argument
110 vout[gmem_idx + HS_SLAB_THREADS * row_idx] = reg
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dimage_ops.cc340 xla::XlaOp row_idx = values[0]; in operator ()() local
342 xla::Lt(row_idx, xla::ConstantR0<int32>(cond_builder, num_boxes)); in operator ()()
361 auto row_idx = values[0]; in operator ()() local
371 std::vector<xla::XlaOp> row_idx_vector = {row_idx}; in operator ()()
379 auto row_iou = xla::DynamicSlice(iou_mask, {row_idx, zero}, {1, num_boxes}); in operator ()()
384 {zero, row_idx}); in operator ()()
391 row_idx = row_idx + xla::ConstantR0<int32>(builder, 1); in operator ()()
392 return std::vector<xla::XlaOp>{row_idx, num_outputs_so_far, iou_mask, in operator ()()
/external/skia/src/compute/hs/cuda/sm_35/
Dhs_cuda_macros.h201 #define HS_SLAB_GLOBAL_LOAD(extent,row_idx) \ argument
202 extent[gmem_idx + HS_SLAB_THREADS * row_idx]
204 #define HS_SLAB_GLOBAL_STORE(row_idx,reg) \ argument
205 vout[gmem_idx + HS_SLAB_THREADS * row_idx] = reg
/external/skqp/src/compute/hs/vk/
Dhs_glsl_macros.h120 #define HS_SLAB_GLOBAL_LOAD(extent,row_idx) \ argument
121 extent[gmem_idx + HS_SLAB_THREADS * row_idx]
123 #define HS_SLAB_GLOBAL_STORE(row_idx,reg) \ argument
124 vout[gmem_idx + HS_SLAB_THREADS * row_idx] = reg
/external/skqp/src/compute/hs/cuda/sm_35/
Dhs_cuda_macros.h201 #define HS_SLAB_GLOBAL_LOAD(extent,row_idx) \ argument
202 extent[gmem_idx + HS_SLAB_THREADS * row_idx]
204 #define HS_SLAB_GLOBAL_STORE(row_idx,reg) \ argument
205 vout[gmem_idx + HS_SLAB_THREADS * row_idx] = reg
/external/skia/src/compute/hs/vk/
Dhs_glsl_macros.h120 #define HS_SLAB_GLOBAL_LOAD(extent,row_idx) \ argument
121 extent[gmem_idx + HS_SLAB_THREADS * row_idx]
123 #define HS_SLAB_GLOBAL_STORE(row_idx,reg) \ argument
124 vout[gmem_idx + HS_SLAB_THREADS * row_idx] = reg
/external/tensorflow/tensorflow/contrib/factorization/python/kernel_tests/
Dmasked_matmul_benchmark.py53 row_idx = random_ops.random_uniform(
57 indices = array_ops.stack([row_idx, col_idx], axis=1)
/external/mesa3d/src/compiler/glsl/
Dast_function.cpp1729 unsigned row_idx = 0; in emit_inline_matrix_constructor() local
1757 unsigned count = MIN2(rows - row_idx, in emit_inline_matrix_constructor()
1762 row_idx, in emit_inline_matrix_constructor()
1768 row_idx += count; in emit_inline_matrix_constructor()
1775 if (row_idx >= rows) { in emit_inline_matrix_constructor()
1776 row_idx = 0; in emit_inline_matrix_constructor()
/external/libaom/libaom/av1/decoder/
Ddecodeframe.c103 for (int row_idx = 1; row_idx < buf->crop_heights[is_uv]; row_idx++) { in set_planes_to_neutral_grey() local
104 memcpy(&base[row_idx * buf->strides[is_uv]], base, in set_planes_to_neutral_grey()
112 for (int row_idx = 0; row_idx < buf->crop_heights[is_uv]; row_idx++) { in set_planes_to_neutral_grey() local
113 memset(&buf->buffers[plane][row_idx * buf->uv_stride], 1 << 7, in set_planes_to_neutral_grey()