Searched defs:new_dims (Results 1 – 11 of 11) sorted by relevance
110 std::vector<int64> new_dims(dims.size() * 2); in Compile() local140 std::vector<int64> new_dims; in Compile() local
30 internal::TransposeDimsVec new_dims; in TestDimensionReduction() local
81 TransposeDimsVec* new_dims) { in ReduceTransposeDimensions()
101 TransposeDimsVec new_dims; in run() local
85 std::vector<int> new_dims; in ReduceGeneric() local
139 int index, absl::Span<const int64> new_dims, in ConstantInputReshaped()
458 DimensionVector new_dims; in StripDegenerateDimensions() local
2219 nvinfer1::Dims new_dims; in ConvertExpandDims() local2280 nvinfer1::Dims new_dims; in ConvertSqueeze() local2479 nvinfer1::Dims new_dims; in ConvertStridedSliceHelper() local
1485 int64_t new_dims[] = {3, 2}; in TEST() local
297 TF_Tensor* to, const int64_t* new_dims, in TF_TensorBitcastFrom()
1535 PartialTensorShape new_dims; in IsSimplifiableReshape() local