Searched refs:shape_values (Results 1 – 2 of 2) sorted by relevance
85 std::vector<int32> shape_values( in InferFilterRank() local88 reinterpret_cast<char*>(shape_values.data())); in InferFilterRank()89 CHECK_EQ(shape_values.size(), 3); in InferFilterRank()92 CHECK_EQ(shape_values[2], -1); in InferFilterRank()93 return shape_values[1]; in InferFilterRank()
446 def _total_size(shape_values): argument453 for val in shape_values: