Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Delemental_ir_emitter.cc2715 int accumulators_count = 1; in EmitElementalReduce() local
2718 accumulators_count = out_shape.tuple_shapes_size(); in EmitElementalReduce()
2726 for (int i = 0; i < accumulators_count; i++) { in EmitElementalReduce()
2779 for (int i = 0; i < accumulators_count; i++) { in EmitElementalReduce()
2790 CHECK(results.size() == accumulators_count); in EmitElementalReduce()
2791 for (int i = 0; i < accumulators_count; i++) { in EmitElementalReduce()