Searched defs:broadcasted_dims (Results 1 – 4 of 4) sorted by relevance
198 std::vector<int64> broadcasted_dims, int64_t last_warp_dim, in BoundSamples()282 std::vector<int64> broadcasted_dims(warp_dims.size() - 1); in CalculateGradData() local583 std::vector<int64> broadcasted_dims(warp_dims.size() - 1); in Compile() local668 std::vector<int64> broadcasted_dims(warp_dims.size() - 1); in Compile() local
133 std::vector<int64> broadcasted_dims; in Compile() local
574 std::vector<int64> broadcasted_dims(num_dims - 1); in ComputeToleranceComparison() local
483 nvinfer1::Dims broadcasted_dims) { in BroadcastWeights()494 nvinfer1::Dims broadcasted_dims, in ApplyBroadcast()