Searched refs:dot_type (Results 1 – 2 of 2) sorted by relevance
| /external/mesa3d/src/gallium/state_trackers/clover/util/ |
| D | algebra.hpp | 138 using dot_type = typename std::common_type< typedef
153 detail::dot_type<U, V>
155 return fold(plus(), detail::dot_type<U, V>(), in dot()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dot_decomposer.cc | 88 PrimitiveType dot_type = dot_shape.element_type(); in DecomposeBatchDot() local
91 Shape dot_shape_r2 = ShapeUtil::MakeShape(dot_type, {dot_rows, dot_cols}); in DecomposeBatchDot()
92 Shape dot_shape_r3 = ShapeUtil::MakeShape(dot_type, {1, dot_rows, dot_cols}); in DecomposeBatchDot()
94 ShapeUtil::MakeShape(dot_type, {batch_size, dot_rows, dot_cols}); in DecomposeBatchDot()
|