Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dload_and_remap_matrix_op.cc65 OP_REQUIRES_OK(context, context->GetAttr("num_cols", &num_cols_)); in LoadAndRemapMatrixOp()
117 context, col_remapping.size() == num_cols_, in Compute()
120 " instead of being equal to num_cols=", num_cols_))); in Compute()
125 col_id_present.resize(num_cols_, true); in Compute()
167 OP_REQUIRES(context, num_cols_ == tensor_shape.dim_size(1), in Compute()
173 " instead of being equal to num_cols=", num_cols_))); in Compute()
203 TensorShape({num_rows_, num_cols_}), in Compute()
260 new_row < num_rows_ && new_col < num_cols_ && in Compute()
265 " and num_cols_=", num_cols_, in Compute()
286 for (int j = 0; j < num_cols_; ++j) { in Compute()
[all …]
Dtopk_op_gpu.h422 SegmentOffsetCreator(int num_cols) : num_cols_(num_cols) {} in SegmentOffsetCreator()
425 return idx * num_cols_; in operator()
428 int num_cols_; member
432 ColumnIndexCreator(int num_cols) : num_cols_(num_cols) {} in ColumnIndexCreator()
436 return ix[0] % num_cols_; in operator()
439 int num_cols_; member
Dsoftmax_op_gpu.cu.cc167 : 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/
Dquery_result_serializer.cc49 : 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()
268 for (uint32_t c = 0; c < num_cols_; c++) in SerializeMetadata()
Dquery_result_serializer.h84 const uint32_t num_cols_; variable
/external/perfetto/src/trace_processor/sqlite/
Dsqlite_vtable_benchmark.cc68 : num_cols_(num_cols), in BenchmarkCursor()
83 size_t num_cols_ = 0; member in __anon85ac22f00111::BenchmarkCursor
97 for (size_t col = 0; col < num_cols_; col++) { in RandomFill()