Searched refs:dims_actual (Results 1 – 2 of 2) sorted by relevance
687 std::vector<DimensionHandle> dims_actual; in MakeShape() local688 dims_actual.reserve(dims.size()); in MakeShape()690 dims_actual.push_back(MakeDim(d)); in MakeShape()693 return shape_manager_.MakeShape(dims_actual); in MakeShape()
502 std::vector<DimensionHandle> dims_actual(num_dims); in MakeShapeFromFormat() local503 dims_actual[GetTensorBatchDimIndex(num_dims, format)] = context->MakeDim(N); in MakeShapeFromFormat()505 dims_actual[outer_c_index] = context->MakeDim(C); in MakeShapeFromFormat()507 dims_actual[GetTensorInnerFeatureDimIndex(num_dims, format)] = in MakeShapeFromFormat()510 dims_actual[GetTensorInnerWidthDimIndex(num_dims, format)] = in MakeShapeFromFormat()515 dims_actual[GetTensorSpatialDimIndex(num_dims, format, spatial_dim)] = in MakeShapeFromFormat()518 *out = context->MakeShape(dims_actual); in MakeShapeFromFormat()