Searched refs:dims_actual (Results 1 – 2 of 2) sorted by relevance
333 std::vector<DimensionHandle> dims_actual(num_dims); in MakeShapeFromFormat() local334 dims_actual[GetTensorBatchDimIndex(num_dims, format)] = context->MakeDim(N); in MakeShapeFromFormat()336 dims_actual[outer_c_index] = context->MakeDim(C); in MakeShapeFromFormat()338 dims_actual[GetTensorInnerFeatureDimIndex(num_dims, format)] = in MakeShapeFromFormat()341 dims_actual[GetTensorInnerWidthDimIndex(num_dims, format)] = in MakeShapeFromFormat()345 dims_actual[GetTensorSpatialDimIndex(num_dims, format, spatial_dim)] = in MakeShapeFromFormat()348 *out = context->MakeShape(dims_actual); in MakeShapeFromFormat()
738 std::vector<DimensionHandle> dims_actual; in MakeShape() local739 dims_actual.reserve(dims.size()); in MakeShape()741 dims_actual.push_back(MakeDim(d)); in MakeShape()744 return shape_manager_.MakeShape(dims_actual); in MakeShape()