| /external/tensorflow/tensorflow/core/kernels/ |
| D | substr_op.cc | 77 tensorflow::internal::SubtleMustCopy(pos_tensor.scalar<T>()()); in Compute()
79 tensorflow::internal::SubtleMustCopy(len_tensor.scalar<T>()()); in Compute()
107 const T pos = tensorflow::internal::SubtleMustCopy(pos_flat(i)); in Compute()
108 const T len = tensorflow::internal::SubtleMustCopy(len_flat(i)); in Compute()
177 const T pos = tensorflow::internal::SubtleMustCopy(pos_bcast(i)); in Compute()
178 const T len = tensorflow::internal::SubtleMustCopy(len_bcast(i)); in Compute()
233 tensorflow::internal::SubtleMustCopy(pos_bcast(i, j)); in Compute()
235 tensorflow::internal::SubtleMustCopy(len_bcast(i, j)); in Compute()
|
| D | segment_reduction_ops_impl.h | 108 ? internal::SubtleMustCopy(segment_vec(num_indices - 1)) + 1 in Compute()
134 Index out_index = internal::SubtleMustCopy(segment_vec(start)); in Compute()
145 next_index = internal::SubtleMustCopy(segment_vec(end)); in Compute()
332 Index j = internal::SubtleMustCopy(segment_ids(i));
401 const int64 output_rows = internal::SubtleMustCopy(static_cast<int64>(
461 output_rows = internal::SubtleMustCopy(num_segments.scalar<int32>()());
484 ? internal::SubtleMustCopy(segment_vec(num_indices - 1)) + 1
522 SegmentId out_index = internal::SubtleMustCopy(segment_vec(start));
530 next_index = internal::SubtleMustCopy(segment_vec(end));
862 const SegmentId M = internal::SubtleMustCopy(output_dim0.scalar<int32>()());
[all …]
|
| D | scatter_functor.h | 159 const Index index = ::tensorflow::internal::SubtleMustCopy(indices(i));
192 const Index index = ::tensorflow::internal::SubtleMustCopy(indices(i));
249 const Index index = ::tensorflow::internal::SubtleMustCopy(indices(i));
285 const Index index = ::tensorflow::internal::SubtleMustCopy(indices(i));
297 const Index index = ::tensorflow::internal::SubtleMustCopy(indices(i));
334 const Index index = ::tensorflow::internal::SubtleMustCopy(indices(i));
359 const Index index = ::tensorflow::internal::SubtleMustCopy(indices(i));
394 const Index index = ::tensorflow::internal::SubtleMustCopy(indices(i));
|
| D | dynamic_partition_op.cc | 60 const int32 p = internal::SubtleMustCopy(e_partitions(i)); in ValidateAndAllocateOutputs()
113 const int32 p = internal::SubtleMustCopy(e_partitions(i)); in Compute()
141 const int32 p = internal::SubtleMustCopy(e_partitions(i)); in Compute()
|
| D | sparse_tensor_dense_matmul_op.cc | 270 const Tindices m = internal::SubtleMustCopy(a_indices(i, lhs_index_a)); in Compute()
271 const Tindices k = internal::SubtleMustCopy(a_indices(i, rhs_index_a)); in Compute()
290 const Tindices m = internal::SubtleMustCopy(a_indices(i, lhs_index_a)); \ in Compute()
291 const Tindices k = internal::SubtleMustCopy(a_indices(i, rhs_index_a)); \ in Compute()
|
| D | stateless_random_ops.cc | 39 seed0 = internal::SubtleMustCopy(seed_vals(0)); in GenerateKey()
40 seed1 = internal::SubtleMustCopy(seed_vals(1)); in GenerateKey()
43 seed0 = internal::SubtleMustCopy(seed_vals(0)); in GenerateKey()
44 seed1 = internal::SubtleMustCopy(seed_vals(1)); in GenerateKey()
|
| D | one_hot_op.h | 98 const TI depth = internal::SubtleMustCopy(indices(i, 0));
110 const TI depth = internal::SubtleMustCopy(indices(d0, d1));
|
| D | concat_op.cc | 92 internal::SubtleMustCopy(concat_dim_tensor.scalar<int32>()()); in Compute()
95 internal::SubtleMustCopy(concat_dim_tensor.scalar<int64>()()); in Compute()
284 const int64 cdim = internal::SubtleMustCopy(concat_dim.scalar<int32>()()); in Compute()
|
| D | spacetobatch_functor.h | 58 (*output)[i] = SubtleMustCopy(eigen_vec(i)); in SubtleMustCopyFlatHelper()
|
| D | sparse_xent_op.h | 73 const Index label = tensorflow::internal::SubtleMustCopy(labels_(batch)); in operator()
113 const Index label = tensorflow::internal::SubtleMustCopy(labels_(batch)); in operator()
|
| D | in_topk_op.h | 72 auto target = internal::SubtleMustCopy(targets(batch_idx));
|
| D | unique_op.cc | 118 axis = internal::SubtleMustCopy(axis_tensor.scalar<int32>()()); in Compute()
120 axis = internal::SubtleMustCopy(axis_tensor.scalar<int64>()()); in Compute()
|
| D | dynamic_stitch_op.cc | 266 int32 index = internal::SubtleMustCopy(indices_vec(i)); in Compute()
278 int32 index = internal::SubtleMustCopy(indices_vec(i)); in Compute()
|
| D | sparse_tensor_dense_add_op.cc | 134 idx[d] = internal::SubtleMustCopy(indices(i, d)); in operator ()()
|
| D | sparse_dense_binary_op_shared.cc | 131 idx[d] = internal::SubtleMustCopy(indices_mat(i, d)); \ in Compute()
|
| D | gather_nd_op_cpu_impl.h | 59 const Index ix_i = internal::SubtleMustCopy(Tindices_(loc, i)); in GenerateIndices()
|
| D | gather_functor.h | 81 const Index index = internal::SubtleMustCopy(indices(indices_idx)); in HandleCopies()
|
| D | scan_ops.cc | 53 internal::SubtleMustCopy(tensor_axis.scalar<Tidx>()()); in Compute()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | image_resizer_state.h | 98 out_height = internal::SubtleMustCopy(Svec(0)); in ValidateAndCalculateOutputSize()
99 out_width = internal::SubtleMustCopy(Svec(1)); in ValidateAndCalculateOutputSize()
|
| D | strided_slice_op.cc | 135 dense->begin[full_index] = internal::SubtleMustCopy<T>(begin_flat[i]); in BuildDenseSpec()
138 dense->end[full_index] = internal::SubtleMustCopy<T>(end_flat[i]); in BuildDenseSpec()
141 internal::SubtleMustCopy<T>(strides_flat[i]); in BuildDenseSpec()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | bounds_check.h | 45 EIGEN_ALWAYS_INLINE EIGEN_DEVICE_FUNC const T SubtleMustCopy(const T &x) { in SubtleMustCopy() function
|
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | crop_and_resize_op.cc | 166 const int crop_height = internal::SubtleMustCopy(crop_size_vec(0)); in ComputeAsync()
167 const int crop_width = internal::SubtleMustCopy(crop_size_vec(1)); in ComputeAsync()
399 const int batch_size = internal::SubtleMustCopy(image_size_vec(0)); in ComputeAsync()
400 const int image_height = internal::SubtleMustCopy(image_size_vec(1)); in ComputeAsync()
401 const int image_width = internal::SubtleMustCopy(image_size_vec(2)); in ComputeAsync()
402 const int depth = internal::SubtleMustCopy(image_size_vec(3)); in ComputeAsync()
|
| D | decode_image_op.cc | 547 int32 header_size_ = internal::SubtleMustCopy( in DecodeBmpV2()
550 int32 width_ = internal::SubtleMustCopy( in DecodeBmpV2()
553 int32 height_ = internal::SubtleMustCopy( in DecodeBmpV2()
556 int16 bpp_ = internal::SubtleMustCopy( in DecodeBmpV2()
|
| D | sample_distorted_bounding_box_op.cc | 268 const uint64 height_raw = internal::SubtleMustCopy(image_size.flat<T>()(0)); in DoCompute()
269 const uint64 width_raw = internal::SubtleMustCopy(image_size.flat<T>()(1)); in DoCompute()
|
| /external/tensorflow/tensorflow/core/util/sparse/ |
| D | sparse_tensor.h | 372 const int64 index = internal::SubtleMustCopy(ix_ptr[n]); in ToDense()
383 const int64 row_index = internal::SubtleMustCopy(ix_ptr[n * 2]); in ToDense()
384 const int64 col_index = internal::SubtleMustCopy(ix_ptr[n * 2 + 1]); in ToDense()
407 const int64 ix_n_d = internal::SubtleMustCopy(ix_ptr[n * dims_ + d]); in ToDense()
|