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 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::__anon47412fb60111::GemvConfig
135 has_addend() ? "_with_addend" : ""); in GetCacheKey()
140 int64 tile_cols, int64 m, int64 k, bool has_addend) in GemvConfig() argument
147 has_addend_(has_addend) {} in GemvConfig()
226 int64 m, int64 k, bool has_addend) in Config() argument
229 /*k=*/k, /*has_addend=*/has_addend) {} in Config()
245 CHECK(!has_addend() || addend != nullptr); in ColumnMajorMatrixVectorProductEmitter()
452 int64 m, int64 k, bool has_addend) in Config() argument
455 /*k=*/k, /*has_addend=*/has_addend) {} in Config()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | dot_operation_test.cc | 265 bool has_addend; member
272 if (param.has_addend) { in PrintDotTestParam()
344 if (param.has_addend) { in TestImpl()
366 if (param.has_addend) { in TestImpl()
377 if (param.has_addend) { in TestImpl()
387 if (param.has_addend) { in TestImpl()
454 for (bool has_addend : {true, false}) { in CreateNoLayoutAssignmentDotTestParameters()
459 /*has_addend=*/has_addend, in CreateNoLayoutAssignmentDotTestParameters()
465 /*has_addend=*/has_addend, in CreateNoLayoutAssignmentDotTestParameters()
|