| /external/tensorflow/tensorflow/core/kernels/ |
| D | scatter_op.cc | 131 const Index bad_i = functor(c, c->template eigen_device<Device>(), in DoCompute() local
133 OP_REQUIRES(c, bad_i < 0, in DoCompute()
135 "indices", SliceDebugString(indices.shape(), bad_i), in DoCompute()
136 " = ", indices_flat(bad_i), " is not in [0, ", in DoCompute()
143 const Index bad_i = functor(c, c->template eigen_device<Device>(), in DoCompute() local
145 OP_REQUIRES(c, bad_i < 0, in DoCompute()
147 "indices", SliceDebugString(indices.shape(), bad_i), in DoCompute()
148 " = ", indices_flat(bad_i), " is not in [0, ", in DoCompute()
219 const Index bad_i = functor(c, c->template eigen_device<SYCLDevice>(), in DoCompute() local
221 OP_REQUIRES(c, bad_i < 0, in DoCompute()
[all …]
|
| D | gather_op.cc | 111 int64 bad_i = functor(c, params_flat, indices_flat, out_flat); in Compute() local
114 c, bad_i < 0, in Compute()
116 "indices", SliceDebugString(indices.shape(), bad_i), " = ", in Compute()
117 indices_flat(bad_i), " is not in [0, ", gather_dim_size, ")")); in Compute()
|
| D | gather_nd_op.cc | 156 Index bad_i = -1; in DoGatherNd() local
170 bad_i = func(c->eigen_device<Device>(), slice_size, scratch_scalar, \ in DoGatherNd()
190 if (bad_i >= 0) { in DoGatherNd()
194 "indices", SliceDebugString(shape, bad_i), " = [", in DoGatherNd()
196 gtl::ArraySlice<Index>(&indices_mat(bad_i, 0), indices_nd), ", "), in DoGatherNd()
|
| D | gather_functor.h | 120 int64 bad_i; in operator() local
128 bad_i = HandleCopies<T, Index, int64, elems>(ctx, params, indices, \ in operator()
132 bad_i = HandleCopies<T, Index, int32, elems>(ctx, params, indices, \ in operator()
145 return bad_i; in operator()
|
| D | resource_variable_ops.cc | 694 int64 bad_i = functor(c, params_flat, indices_flat, out_flat); in Compute() local
697 c, bad_i < 0, in Compute()
699 "indices", SliceDebugString(indices.shape(), bad_i), " = ", in Compute()
700 indices_flat(bad_i), " is not in [0, ", params.dim_size(0), ")")); in Compute()
791 const Index bad_i = functor(c, c->template eigen_device<Device>(), in Compute() local
793 OP_REQUIRES(c, bad_i < 0, in Compute()
795 "indices", SliceDebugString(indices.shape(), bad_i), in Compute()
796 " = ", indices_flat(bad_i), " is not in [0, ", in Compute()
808 const Index bad_i = functor(c, c->template eigen_device<Device>(), in Compute() local
810 OP_REQUIRES(c, bad_i < 0, in Compute()
[all …]
|
| D | scatter_nd_op.cc | 693 Index bad_i = -1; in DoScatterNd() local
704 bad_i = \ in DoScatterNd()
725 if (bad_i >= 0) { in DoScatterNd()
729 "indices", SliceDebugString(slice_shape, bad_i), " = [", in DoScatterNd()
731 gtl::ArraySlice<Index>(&indices_flat(bad_i, 0), slice_dim), ", "), in DoScatterNd()
|