Searched defs:broadcasted_dims (Results 1 – 6 of 6) sorted by relevance
201 std::vector<int64_t> broadcasted_dims, int64_t last_warp_dim, in BoundSamples()285 std::vector<int64_t> broadcasted_dims(warp_dims.size() - 1); in CalculateGradData() local584 std::vector<int64_t> broadcasted_dims(warp_dims.size() - 1); in Compile() local666 std::vector<int64_t> broadcasted_dims(warp_dims.size() - 1); in Compile() local
133 std::vector<int64_t> broadcasted_dims; in Compile() local
98 nvinfer1::Dims broadcasted_dims[2]; local
450 std::vector<int64_t> broadcasted_dims(shape_params.rank); in TridiagonalMatMul() local
575 std::vector<int64_t> broadcasted_dims(num_dims - 1); in ComputeToleranceComparison() local
407 const DimsAdapter& broadcasted_dims) { in BroadcastWeights()418 const DimsAdapter& broadcasted_dims, in ApplyBroadcast()