Searched defs:numInputDims (Results 1 – 6 of 6) sorted by relevance
248 int numInputDims = input.dim(); in replication_pad2d_backward_out_cuda_template() local312 int numInputDims = input.dim(); in shapeAndGradOutputCheck3d() local378 int numInputDims = input.dim(); in replication_pad3d_backward_out_cuda_template() local430 int numInputDims = input.ndimension(); in TORCH_IMPL_FUNC() local491 int64_t numInputDims = input.ndimension(); in TORCH_IMPL_FUNC() local617 int numInputDims = input.dim(); in TORCH_IMPL_FUNC() local
59 int32_t numInputDims = static_cast<int32_t>(getNumberOfDimensions(inputShape)); in compute() local100 uint32_t numInputDims = getNumberOfDimensions(inputShape); in prepare() local
42 int32_t numInputDims = static_cast<int32_t>(getNumberOfDimensions(inputShape)); in prepare() local
80 uint32_t numInputDims = getNumberOfDimensions(input); in prepare() local
100 int32_t numInputDims = static_cast<int32_t>(getNumberOfDimensions(inputShape)); in padGeneric() local
412 uint32_t numInputDims = getNumberOfDimensions(input); in padPrepare() local507 int32_t numInputDims = static_cast<int32_t>(getNumberOfDimensions(input)); in meanPrepare() local