diff options
Diffstat (limited to 'tensorflow/core/kernels/dynamic_stitch_op.cc')
| -rw-r--r-- | tensorflow/core/kernels/dynamic_stitch_op.cc | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/core/kernels/dynamic_stitch_op.cc b/tensorflow/core/kernels/dynamic_stitch_op.cc index b01db91720..fb2a4cc8ef 100644 --- a/tensorflow/core/kernels/dynamic_stitch_op.cc +++ b/tensorflow/core/kernels/dynamic_stitch_op.cc @@ -247,8 +247,8 @@ class DynamicStitchOpImplCPU : public DynamicStitchOpImplBase<T> { data.shaped<T, 2>({indices_vec.dimension(0), slice_size}); if (DataTypeCanUseMemcpy(DataTypeToEnum<T>::v())) { - T* merged_base = &merged_flat(0, 0); - const T* data_base = &data_flat(0, 0); + T* merged_base = merged_flat.data(); + const T* data_base = data_flat.data(); for (int i = 0; i < indices_vec.size(); i++) { int32 index = internal::SubtleMustCopy(indices_vec(i)); OP_REQUIRES( |