Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_cost_analysis.cc329 int64 reduction_width = 1; in HandleDot() local
331 reduction_width *= lhs_shape.dimensions(dim); in HandleDot()
335 kFmaFlops * ShapeUtil::ElementsIn(dot_shape) * reduction_width; in HandleDot()