Searched refs:input0_rank (Results 1 – 1 of 1) sorted by relevance
57 const int32 input0_rank = c->Rank(input0_shape); in __anon9b840f3f0102() local60 c->WithRank(input1_shape, input0_rank, &input1_shape)); in __anon9b840f3f0102()74 output_rank = c->MakeDim(input0_rank); in __anon9b840f3f0102()119 const int32 input0_rank = c->Rank(input0_shape); in __anon9b840f3f0202() local121 c->WithValue(input1_rank_dim, input0_rank, &input1_rank_dim)); in __anon9b840f3f0202()122 output_rank_dim = c->MakeDim(input0_rank); in __anon9b840f3f0202()166 const int64 input0_rank = c->Value(input0_rank_dim); in __anon9b840f3f0302() local167 if (input0_rank < 2) { in __anon9b840f3f0302()169 input0_rank, "."); in __anon9b840f3f0302()172 c->WithValue(input1_rank_dim, input0_rank, &input1_rank_dim)); in __anon9b840f3f0302()