Searched refs:dim_vec (Results 1 – 3 of 3) sorted by relevance
179 std::vector<nvinfer1::Dims3> dim_vec(2, dims); in TEST_P() local180 std::vector<TensorShape> shape_vec = DimVecToShapeVec(dim_vec); in TEST_P()205 for (auto dim_vec : input_profiles) { in TEST_P() local206 std::vector<TensorShape> shape_vec = DimVecToShapeVec(dim_vec, true); in TEST_P()
230 std::vector<tensorflow::shape_inference::DimensionHandle> dim_vec; in ShapeHandleFromDims() local231 dim_vec.reserve(num_dims); in ShapeHandleFromDims()233 dim_vec.push_back(ic->MakeDim(dims[i])); in ShapeHandleFromDims()235 return ic->MakeShape(dim_vec); in ShapeHandleFromDims()
3795 dim_vec = %dim.as_constant()3796 for(int32 i = 0 i < dim_vec.size() i++) {3797 total_size *= dim_vec[i]3800 output_type = tensor<dim_vec, filled_val.dtype>