Searched defs:data_rank (Results 1 – 11 of 11) sorted by relevance
68 const int data_rank = ValueRank(unsorted_segmented_sum.data()); in ComputeLayoutBackward() local103 const int data_rank = data_layout.rank(); in ExpandOp() local
37 def _makeShapeTuple(self, batch_size, channel_count, data_rank, data_dim, argument74 def testDeterministicGradients(self, data_layout, data_rank, data_type): argument
266 int64_t data_rank; member
866 const int64_t data_rank = scatter.scatter_updates()[0]->shape().rank(); in ScatterEffectiveDataSharding() local1327 int64_t data_rank) { in ReplicateAllDataDims()
949 const int64_t data_rank = partial_rep.TiledDataRank(); in RefineManualAutoShardingFromAuto() local
55 const int data_rank = NumDimensions(data); in ResizeOutputTensor() local
56 const int data_rank = NumDimensions(data); in ResizeOutputTensor() local
88 const int data_rank = tflite::NumDimensions(data); in ResizeOutputTensor() local
856 const int32_t data_rank = c->Rank(data); in __anon247ac1530402() local
922 int data_rank; in verify() local
5709 int64_t data_rank = data_type.getRank(); in matchAndRewrite() local