Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dsparse_matmul_op.cc275 const int kNumOperands = (sizeof(Packet) / sizeof(float)); variable
351 if (kNumOperands >= 2) { in LoadTwoScalars()
364 if (kNumOperands >= 4) { in LoadFourScalars()
414 *binp += 2 * kNumOperands; in MulAdd()
416 auto c2 = LOAD(*out + kNumOperands); in MulAdd()
420 STORE(*out + kNumOperands, c2); in MulAdd()
421 *out += 2 * kNumOperands; in MulAdd()
432 auto c2 = LOAD(*out + kNumOperands); in MulAdd3Way()
436 *binp1 += 2 * kNumOperands; in MulAdd3Way()
440 *binp2 += 2 * kNumOperands; in MulAdd3Way()
[all …]