Searched refs:num_cols_ (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | load_and_remap_matrix_op.cc | 65 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()
157 OP_REQUIRES(context, num_cols_ == tensor_shape.dim_size(1), in Compute()
163 " instead of being equal to num_cols=", num_cols_))); in Compute()
193 TensorShape({num_rows_, num_cols_}), in Compute()
250 new_row < num_rows_ && new_col < num_cols_ && in Compute()
255 " and num_cols_=", num_cols_, in Compute()
276 for (int j = 0; j < num_cols_; ++j) { in Compute()
[all …]
|
| D | topk_op_gpu.h | 430 SegmentOffsetCreator(int num_cols) : num_cols_(num_cols) {}
433 return idx * num_cols_;
436 int num_cols_;
440 ColumnIndexCreator(int num_cols) : num_cols_(num_cols) {}
444 return ix[0] % num_cols_;
447 int num_cols_;
|
| D | softmax_op_gpu.cu.cc | 167 : logits_(logits), max_logits_(max_logits), num_cols_(num_cols) {} in SubtractAndExpFunctor()
172 strict_cast<U>(logits_[gid] - ldg(max_logits_ + gid / num_cols_)); in operator ()()
178 const int num_cols_; member
|
| /external/perfetto/src/trace_processor/rpc/ |
| D | query_result_serializer.cc | 49 : iter_(iter.take_impl()), num_cols_(iter_->ColumnCount()) {} in QueryResultSerializer()
120 if (col_ >= num_cols_) { in SerializeBatch()
128 PERFETTO_DCHECK(num_cols_ > 0); in SerializeBatch()
132 if (cell_idx + num_cols_ > cells_per_batch_ || in SerializeBatch()
264 for (uint32_t c = 0; c < num_cols_; c++) in SerializeColumnNames()
|
| D | query_result_serializer.h | 83 const uint32_t num_cols_; variable
|
| /external/perfetto/src/trace_processor/sqlite/ |
| D | sqlite_vtable_benchmark.cc | 52 : num_cols_(num_cols), batch_size_(batch_size), rnd_engine_(kRandomSeed) { in BenchmarkCursor()
64 size_t num_cols_; member in __anon036853120111::BenchmarkCursor
76 for (size_t col = 0; col < num_cols_; col++) { in RandomFill()
|