Searched refs:has_addend (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | tiled_dot_emitter.cc | 119 int64_t has_addend() const { return derived().config().has_addend(); } in has_addend() function
130 bool has_addend() const { return has_addend_; } in has_addend() function in xla::cpu::__anon8d179a540111::GemvConfig
135 has_addend() ? "_with_addend" : ""); in GetCacheKey()
141 int64_t k, bool has_addend) in GemvConfig() argument
148 has_addend_(has_addend) {} in GemvConfig()
227 int64_t tile_cols, int64_t m, int64_t k, bool has_addend) in Config() argument
230 /*k=*/k, /*has_addend=*/has_addend) {} in Config()
247 CHECK(!has_addend() || addend != nullptr); in ColumnMajorMatrixVectorProductEmitter()
455 int64_t tile_cols, int64_t m, int64_t k, bool has_addend) in Config() argument
458 /*k=*/k, /*has_addend=*/has_addend) {} in Config()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | dot_operation_test.cc | 264 bool has_addend; member
271 if (param.has_addend) { in PrintDotTestParam()
343 if (param.has_addend) { in TestImpl()
365 if (param.has_addend) { in TestImpl()
376 if (param.has_addend) { in TestImpl()
386 if (param.has_addend) { in TestImpl()
453 for (bool has_addend : {true, false}) { in CreateNoLayoutAssignmentDotTestParameters()
458 /*has_addend=*/has_addend, in CreateNoLayoutAssignmentDotTestParameters()
464 /*has_addend=*/has_addend, in CreateNoLayoutAssignmentDotTestParameters()
|