saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-30 20:18:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Remove unnecessary parameter in some places in GatherElements implementation (#2612)

Browse source 
* Remove unnecessary parameter in some places

* Update

* Update
This commit is contained in:
Hariharan Seshadri 2019-12-11 11:15:12 -08:00 committed by GitHub
parent 8729784635
commit b2d65b447f
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 0 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
onnxruntime/core/providers/cuda/tensor/gather_elements.cc
View file

@ -27,7 +27,6 @@ Status GatherElements::ComputeInternal(OpKernelContext* context) const {
const auto& input_shape = input_tensor->Shape();
const auto& input_dims = input_shape.GetDims();
const int64_t input_rank = static_cast<int64_t>(input_dims.size());
const int64_t input_size = input_shape.Size();
// Process indices tensor
const auto* indices_tensor = context->Input<Tensor>(1);
@ -67,7 +66,6 @@ Status GatherElements::ComputeInternal(OpKernelContext* context) const {
GatherElementsImpl<int32_t>(
input_rank,
input_tensor->DataRaw(),
input_size,
input_dims[axis],
gpu_input_strides.GpuPtr(),
indices_data,
@ -82,7 +80,6 @@ Status GatherElements::ComputeInternal(OpKernelContext* context) const {
GatherElementsImpl<int64_t>(
input_rank,
input_tensor->DataRaw(),
input_size,
input_dims[axis],
gpu_input_strides.GpuPtr(),
indices_data,

3
onnxruntime/core/providers/cuda/tensor/gather_elements_impl.cu
View file

@ -42,7 +42,6 @@ template <typename Tin>
void GatherElementsImpl(
const int64_t rank,
const void* input_data,
const int64_t input_size,
const int64_t input_dim_along_axis,
const int64_t* input_strides,
const Tin* indices_data,
@ -95,7 +94,6 @@ void GatherElementsImpl(
template void GatherElementsImpl<int32_t>(
const int64_t rank,
const void* input_data,
const int64_t input_size,
const int64_t input_dim_along_axis,
const int64_t* input_strides,
const int32_t* indices_data,
@ -108,7 +106,6 @@ template void GatherElementsImpl<int32_t>(
template void GatherElementsImpl<int64_t>(
const int64_t rank,
const void* input_data,
const int64_t input_size,
const int64_t input_dim_along_axis,
const int64_t* input_strides,
const int64_t* indices_data,

1
onnxruntime/core/providers/cuda/tensor/gather_elements_impl.h
View file

@ -13,7 +13,6 @@ template <typename Tin>
void GatherElementsImpl(
const int64_t rank, // both inputs have same rank and this is validated in the main Compute
const void* input_data,
const int64_t input_size,
const int64_t input_dim_along_axis,
const int64_t* input_strides,
const Tin* indices_data,