Searched refs:dim_vec (Results 1 – 2 of 2) sorted by relevance
68 std::vector<DimensionHandle> dim_vec; in TRTEngineOpShapeInference() local69 dim_vec.emplace_back(context->MakeDim(num_batch)); in TRTEngineOpShapeInference()73 dim_vec.emplace_back(context->MakeDim(dims.d[j])); in TRTEngineOpShapeInference()78 output_shape = context->MakeShape(dim_vec); in TRTEngineOpShapeInference()
664 std::vector<tensorflow::shape_inference::DimensionHandle> dim_vec; in ShapeHandleFromDims() local665 dim_vec.reserve(num_dims); in ShapeHandleFromDims()667 dim_vec.push_back(ic->MakeDim(dims[i])); in ShapeHandleFromDims()669 return ic->MakeShape(dim_vec); in ShapeHandleFromDims()