Searched defs:data_rank (Results 1 – 7 of 7) sorted by relevance
41 def _makeShapeTuple(self, batch_size, channel_count, data_rank, data_dim, argument78 def testDeterministicGradients(self, data_layout, data_rank, data_type): argument
55 const int data_rank = NumDimensions(data); in ResizeOutputTensor() local
310 int64 data_rank; member
786 const int64_t data_rank = hlo.operand(2)->shape().rank(); in ScatterEffectiveDataSharding() local1234 int64_t data_rank = sharding.tile_assignment().num_dimensions(); in PartiallyReplicateTiledShardingOnAllDimsExcept() local
844 const int32_t data_rank = c->Rank(data); in __anon99f049890402() local
881 int data_rank; in Verify() local
5987 int64_t data_rank = data_type.getRank(); in matchAndRewrite() local