Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dspmd_partitioner.cc3507 std::vector<const Shape*> new_operand_shapes(input_count * 2); in HandleReduce() local
3509 new_operand_shapes[i] = &inputs[i].hlo()->shape(); in HandleReduce()
3510 new_operand_shapes[i + input_count] = &inits[i]->shape(); in HandleReduce()
3515 ShapeInference::InferReduceShape(new_operand_shapes, hlo->dimensions(), in HandleReduce()