Home
last modified time | relevance | path

Searched defs:sharded_input (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.h264 HloInstruction* sharded_input; member
Dspmd_partitioner_test.cc862 auto sharded_input = in TEST_F() local
941 auto sharded_input = in TEST_F() local
983 auto sharded_input = in TEST_F() local
1032 auto sharded_input = AllOf(op::Shape("f32[2,2,2,2]{3,2,1,0}"), in TEST_F() local
Dspmd_partitioner.cc759 auto sharded_input = in ReshardAsWindowedInput() local