Searched defs:data_rank (Results 1 – 7 of 7) sorted by relevance
42 def _makeShapeTuple(self, batch_size, channel_count, data_rank, data_dim, argument79 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
823 const int32 data_rank = c->Rank(data); in __anon57f5a5dd0402() local
757 const int64 data_rank = hlo.operand(2)->shape().rank(); in ScatterEffectiveDataSharding() local
755 int data_rank; in Verify() local
4872 int64_t data_rank = data_type.getRank(); in matchAndRewrite() local