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.h304 HloInstruction* sharded_input; member
Dspmd_partitioner_test.cc897 auto sharded_input = in TEST_F() local
975 auto sharded_input = in TEST_F() local
1017 auto sharded_input = in TEST_F() local
1066 auto sharded_input = AllOf(op::Shape("f32[2,2,2,2]{3,2,1,0}"), in TEST_F() local
Dspmd_partitioner.cc992 auto sharded_input = in ReshardAsWindowedInput() local