Home
last modified time | relevance | path

Searched refs:num_cols_ (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/contrib/image/kernels/
Dsegmentation_ops.h68 num_cols_(num_cols), in BlockedImageUnionFindFunctor()
90 return (num_cols_ + block_width_ - 1) / block_width_; in num_blocks_horizontally()
109 return block_height_ < num_rows_ || block_width_ < num_cols_; in can_merge()
130 if (0 <= block_center_x && block_center_x + 1 < num_cols_) { in merge_internal_block_edges()
141 std::min(num_cols_, block_start_x + block_width_); in merge_internal_block_edges()
152 const int64 num_cols_; variable
171 const int64 index_a = col + num_cols_ * (row + num_rows_ * batch); in union_down()
173 const int64 index_b = col + num_cols_ * (row + 1 + num_rows_ * batch); in union_down()
185 const int64 index_a = col + num_cols_ * (row + num_rows_ * batch); in union_right()
186 if (col + 1 < num_cols_ && read_pixel(batch, row, col + 1) == pixel) { in union_right()
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dload_and_remap_matrix_op.cc65 OP_REQUIRES_OK(context, context->GetAttr("num_cols", &num_cols_)); in LoadAndRemapMatrixOp()
112 context, col_remapping.size() == num_cols_, in Compute()
115 " instead of being equal to num_cols=", num_cols_))); in Compute()
120 col_id_present.resize(num_cols_, true); in Compute()
158 OP_REQUIRES(context, num_cols_ == tensor_shape.dim_size(1), in Compute()
164 " instead of being equal to num_cols=", num_cols_))); in Compute()
194 TensorShape({num_rows_, num_cols_}), in Compute()
251 new_row < num_rows_ && new_col < num_cols_ && in Compute()
256 " and num_cols_=", num_cols_, in Compute()
277 for (int j = 0; j < num_cols_; ++j) { in Compute()
[all …]
Dtopk_op_gpu.h412 SegmentOffsetCreator(int num_cols) : num_cols_(num_cols) {}
415 return idx * num_cols_;
418 int num_cols_;
422 ColumnIndexCreator(int num_cols) : num_cols_(num_cols) {}
426 return ix[0] % num_cols_;
429 int num_cols_;
Dsoftmax_op_gpu.cu.cc101 : logits_(logits), max_logits_(max_logits), num_cols_(num_cols) {} in SubtractAndExpFunctor()
106 strict_cast<U>(logits_[gid] - ldg(max_logits_ + gid / num_cols_)); in operator ()()
112 const int num_cols_; member