Searched refs:dims_actual (Results 1 – 2 of 2) sorted by relevance
680 std::vector<DimensionHandle> dims_actual; in MakeShape() local681 dims_actual.reserve(dims.size()); in MakeShape()683 dims_actual.push_back(MakeDim(d)); in MakeShape()686 return shape_manager_.MakeShape(dims_actual); in MakeShape()
499 std::vector<DimensionHandle> dims_actual(num_dims); in MakeShapeFromFormat() local500 dims_actual[GetTensorBatchDimIndex(num_dims, format)] = context->MakeDim(N); in MakeShapeFromFormat()502 dims_actual[outer_c_index] = context->MakeDim(C); in MakeShapeFromFormat()504 dims_actual[GetTensorInnerFeatureDimIndex(num_dims, format)] = in MakeShapeFromFormat()507 dims_actual[GetTensorInnerWidthDimIndex(num_dims, format)] = in MakeShapeFromFormat()512 dims_actual[GetTensorSpatialDimIndex(num_dims, format, spatial_dim)] = in MakeShapeFromFormat()515 *out = context->MakeShape(dims_actual); in MakeShapeFromFormat()