Searched defs:shape_values (Results 1 – 6 of 6) sorted by relevance
193 DataVec shape_values{{"one", {}}, {"two", shape_tensor}}; in CreateTRTEngine() local383 DataVec shape_values{{"one", {}}, {"two", shape_tensor}}; in TEST_P() local
86 std::vector<int32> shape_values( in InferFilterRank() local
46 std::initializer_list<ShapeType> shape_values) { in BroadcastToOpModel()
1084 def _total_size(shape_values): argument
4784 SmallVector<Value, 4> shape_values; in matchAndRewrite() local5570 SmallVector<Value, 4> shape_values; in matchAndRewrite() local
2835 const auto& shape_values = shape.flat<int32>(); in CreateUpdatedShapeProto() local