Searched defs:value_rank (Results 1 – 5 of 5) sorted by relevance
727 const int value_rank = ValueRank(strided_slice_op.value()); in ExpandOp() local791 const int value_rank = ValueRank(strided_slice_op.value()); in ComputeLayoutForward() local827 const int value_rank = ValueRank(strided_slice_op.value()); in ComputeLayoutBackward() local
141 def testDistributeMutableHashTable(self, value_rank): argument
1715 def _compute_all_gather_output_shape(self, value_shape, value_rank, axis): argument
2818 int64_t value_rank = value_type.getRank(); in verify() local
5510 int64_t value_rank = value_type.getRank(); in matchAndRewrite() local5558 int64_t value_rank = value_type.getRank(); in matchAndRewrite() local