Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dreduction_dimension_grouper.cc53 auto is_reduced = [&](int dim) { in HandleReduce() local
65 if (is_reduced(logical_dim) && logical_dim < shape.rank() - 1 && in HandleReduce()
66 is_reduced(logical_dim + 1)) { in HandleReduce()
73 if (is_reduced(logical_dim)) { in HandleReduce()
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dlegalize_to_linalg.cc949 bool is_reduced = llvm::is_contained(reduction_dims, i); in matchAndRewrite() local
950 types.push_back(is_reduced ? getReductionIteratorTypeName() in matchAndRewrite()
954 if (!is_reduced) { in matchAndRewrite()