| /external/tensorflow/tensorflow/core/kernels/ |
| softmax_op_gpu.cu.cc | 61 : logits_(logits), max_logits_(max_logits), num_cols_(num_cols) {}
64 return exp(logits_[gid] - ldg(max_logits_ + gid / num_cols_));
69 const int num_cols_; member in struct:tensorflow::__anon39920::SubtractAndExpFunctor
|
| load_and_remap_matrix_op.cc | 65 OP_REQUIRES_OK(context, context->GetAttr("num_cols", &num_cols_));
112 context, col_remapping.size() == num_cols_,
115 " instead of being equal to num_cols=", num_cols_)));
120 col_id_present.resize(num_cols_, true);
158 OP_REQUIRES(context, num_cols_ == tensor_shape.dim_size(1),
164 " instead of being equal to num_cols=", num_cols_)));
194 TensorShape({num_rows_, num_cols_}),
251 new_row < num_rows_ && new_col < num_cols_ &&
256 " and num_cols_=", num_cols_,
300 int64 num_cols_; member in class:tensorflow::LoadAndRemapMatrixOp
[all...] |
| /external/tensorflow/tensorflow/contrib/image/kernels/ |
| segmentation_ops.h | 68 num_cols_(num_cols),
90 return (num_cols_ + block_width_ - 1) / block_width_;
109 return block_height_ < num_rows_ || block_width_ < num_cols_;
130 if (0 <= block_center_x && block_center_x + 1 < num_cols_) {
141 std::min(num_cols_, block_start_x + block_width_);
152 const int64 num_cols_; member in class:tensorflow::functor::BlockedImageUnionFindFunctor
171 const int64 index_a = col + num_cols_ * (row + num_rows_ * batch);
173 const int64 index_b = col + num_cols_ * (row + 1 + num_rows_ * batch);
185 const int64 index_a = col + num_cols_ * (row + num_rows_ * batch);
186 if (col + 1 < num_cols_ && read_pixel(batch, row, col + 1) == pixel)
[all...] |
