HomeSort by relevance Sort by last modified time
    Searched defs:kRows (Results 1 - 11 of 11) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
dynamic_partition_op_test.cc 176 const int kRows = ((128 << 20) / sizeof(T)) / dim;
177 Tensor data(DataTypeToEnum<T>::value, TensorShape({kRows, dim}));
182 Tensor partitions(DT_INT32, TensorShape({kRows}));
183 for (int i = 0; i < kRows; i++) {
gather_op_test.cc 185 const int kRows = ((512 << 20) / sizeof(float)) / dim;
186 Tensor params(DT_FLOAT, TensorShape({kRows, dim}));
194 indices_vec.push_back(rnd.Uniform(kRows));
scatter_nd_op_test.cc 256 const int kRows = 10000000 / embedding_size;
258 values.reserve(kRows);
259 for (int i = 0; i < kRows * embedding_size; i++) {
268 indices.push_back(rnd.Uniform(kRows));
276 bm.AddInputFromArray<float>(TensorShape({kRows, embedding_size}), values);
cwise_ops_test.cc 39 const int kRows = 100000;
41 int RowsAndColsArg(int r, int c) { return r * kRows + c; }
42 int RowsFromArg(int arg) { return (arg / kRows); }
43 int ColsFromArg(int arg) { return (arg % kRows); }
scatter_op_test.cc 246 const int kRows = 10000000 / embedding_size;
248 values.reserve(kRows);
249 for (int i = 0; i < kRows * embedding_size; i++) {
258 indices.push_back(rnd.Uniform(kRows));
266 bm.AddInputFromArray<float>(TensorShape({kRows, embedding_size}), values);
  /external/gemmlowp/internal/
kernel.h 165 static const int kRows = Lhs::Cell::kWidth * Lhs::kCells;
output.h 44 static constexpr int kRows = InputType::kRows;
51 using OutputType = RegisterBlock<OutputScalarType, kRows, kCols>;
simd_wrappers.h 64 static constexpr int kRows = tRows;
66 static constexpr int kScalarCount = kRows * kCols;
98 (LhsType::kRows < RhsType::kRows));
132 static constexpr int kRows =
133 Lhs::kRows > Rhs::kRows ? Lhs::kRows : Rhs::kRows;
142 using Type = RegisterBlock<ScalarType, Shape::kRows, Shape::kCols>
    [all...]
  /external/skia/gm/
complexclip2.cpp 39 fTotalHeight = kRows * fHeight + SK_Scalar1 * (kRows + 1) * kPadY;
96 for (int i = 0; i < kRows; ++i) {
99 fOps[j*kRows+i][k] = ops[r.nextU() % SK_ARRAY_COUNT(ops)];
105 static constexpr int kRows = 5;
148 for (int i = 0; i < kRows; ++i) {
176 fOps[j*kRows+i][k],
181 fOps[j*kRows+i][k],
186 fOps[j*kRows+i][k],
203 SkClipOp fOps[kRows * kCols][5]
    [all...]
  /external/skqp/gm/
complexclip2.cpp 39 fTotalHeight = kRows * fHeight + SK_Scalar1 * (kRows + 1) * kPadY;
96 for (int i = 0; i < kRows; ++i) {
99 fOps[j*kRows+i][k] = ops[r.nextU() % SK_ARRAY_COUNT(ops)];
105 static constexpr int kRows = 5;
148 for (int i = 0; i < kRows; ++i) {
176 fOps[j*kRows+i][k],
181 fOps[j*kRows+i][k],
186 fOps[j*kRows+i][k],
203 SkClipOp fOps[kRows * kCols][5]
    [all...]
  /external/gemmlowp/standalone/
neon-gemm-kernel-benchmark.cc 239 static const int kRows = Lhs::Cell::kWidth * Lhs::kCells;
    [all...]

Completed in 451 milliseconds