HomeSort by relevance Sort by last modified time
    Searched refs:SubtleMustCopy (Results 1 - 25 of 36) sorted by null

1 2

  /external/tensorflow/tensorflow/core/kernels/
substr_op.cc 63 tensorflow::internal::SubtleMustCopy(pos_tensor.scalar<T>()());
65 tensorflow::internal::SubtleMustCopy(len_tensor.scalar<T>()());
80 const T pos = tensorflow::internal::SubtleMustCopy(pos_flat(i));
81 const T len = tensorflow::internal::SubtleMustCopy(len_flat(i));
146 const T pos = tensorflow::internal::SubtleMustCopy(pos_bcast(i));
147 const T len = tensorflow::internal::SubtleMustCopy(len_bcast(i));
197 tensorflow::internal::SubtleMustCopy(pos_bcast(i, j));
199 tensorflow::internal::SubtleMustCopy(len_bcast(i, j));
bounds_check.h 45 EIGEN_ALWAYS_INLINE EIGEN_DEVICE_FUNC const T SubtleMustCopy(const T &x) {
47 "SubtleMustCopy can only be used on integer types.");
decode_bmp_op.cc 66 int32 header_size_ = internal::SubtleMustCopy(
69 int32 width_ = internal::SubtleMustCopy(
72 int32 height_ = internal::SubtleMustCopy(
75 int32 bpp_ = internal::SubtleMustCopy(
stateless_random_ops.cc 57 seed0 = internal::SubtleMustCopy(seed(0));
58 seed1 = internal::SubtleMustCopy(seed(1));
62 seed0 = internal::SubtleMustCopy(seed(0));
63 seed1 = internal::SubtleMustCopy(seed(1));
spacetobatch_functor.h 58 (*output)[i] = SubtleMustCopy(eigen_vec(i));
scatter_functor.h 147 const Index index = ::tensorflow::internal::SubtleMustCopy(indices(i));
171 const Index index = ::tensorflow::internal::SubtleMustCopy(indices(i));
196 const Index index = ::tensorflow::internal::SubtleMustCopy(indices(i));
207 const Index index = ::tensorflow::internal::SubtleMustCopy(indices(i));
233 const Index index = ::tensorflow::internal::SubtleMustCopy(indices(i));
image_resizer_state.h 71 out_height = internal::SubtleMustCopy(Svec(0));
72 out_width = internal::SubtleMustCopy(Svec(1));
dynamic_partition_op.cc 60 const int32 p = internal::SubtleMustCopy(e_partitions(i));
113 const int32 p = internal::SubtleMustCopy(e_partitions(i));
141 const int32 p = internal::SubtleMustCopy(e_partitions(i));
segment_reduction_ops.cc 97 ? internal::SubtleMustCopy(segment_vec(num_indices - 1)) + 1
123 Index out_index = internal::SubtleMustCopy(segment_vec(start));
134 next_index = internal::SubtleMustCopy(segment_vec(end));
381 Index j = internal::SubtleMustCopy(segment_ids(i));
478 internal::SubtleMustCopy(num_segments.scalar<int32>()());
639 output_rows = internal::SubtleMustCopy(num_segments.scalar<int32>()());
663 ? internal::SubtleMustCopy(segment_vec(num_indices - 1)) + 1
690 OutputRow out_index = internal::SubtleMustCopy(segment_vec(start));
698 next_index = internal::SubtleMustCopy(segment_vec(end));
    [all...]
in_topk_op.cc 76 auto target = internal::SubtleMustCopy(targets(b));
sparse_tensor_dense_matmul_op.cc 268 const Tindices m = internal::SubtleMustCopy(a_indices(i, lhs_index_a));
269 const Tindices k = internal::SubtleMustCopy(a_indices(i, rhs_index_a));
288 const Tindices m = internal::SubtleMustCopy(a_indices(i, lhs_index_a)); \
289 const Tindices k = internal::SubtleMustCopy(a_indices(i, rhs_index_a)); \
sparse_xent_op.h 72 const Index label = tensorflow::internal::SubtleMustCopy(labels_(batch));
112 const Index label = tensorflow::internal::SubtleMustCopy(labels_(batch));
gather_nd_op_cpu_impl.h 59 const Index ix_i = internal::SubtleMustCopy(Tindices_(loc, i));
sparse_dense_binary_op_shared.cc 134 idx[d] = internal::SubtleMustCopy(indices_mat(i, d)); \
sparse_tensor_dense_add_op.cc 134 idx[d] = internal::SubtleMustCopy(indices(i, d));
concat_op.cc 66 internal::SubtleMustCopy(concat_dim_tensor->scalar<int32>()());
297 const int64 cdim = internal::SubtleMustCopy(concat_dim.scalar<int32>()());
scatter_nd_op_cpu_impl.h 119 const Index ix_d = internal::SubtleMustCopy(Tindices(loc, dim));
201 const Index ix_d = internal::SubtleMustCopy(Tindices(loc, dim));
unique_op.cc 74 axis = internal::SubtleMustCopy(axis_tensor.scalar<int32>()());
76 axis = internal::SubtleMustCopy(axis_tensor.scalar<int64>()());
crop_and_resize_op.cc 161 const int crop_height = internal::SubtleMustCopy(crop_size_vec(0));
162 const int crop_width = internal::SubtleMustCopy(crop_size_vec(1));
356 const int batch_size = internal::SubtleMustCopy(image_size_vec(0));
357 const int image_height = internal::SubtleMustCopy(image_size_vec(1));
358 const int image_width = internal::SubtleMustCopy(image_size_vec(2));
359 const int depth = internal::SubtleMustCopy(image_size_vec(3));
dynamic_stitch_op.cc 253 int32 index = internal::SubtleMustCopy(indices_vec(i));
265 int32 index = internal::SubtleMustCopy(indices_vec(i));
sample_distorted_bounding_box_op.cc 257 const uint64 height_raw = internal::SubtleMustCopy(image_size.flat<T>()(0));
258 const uint64 width_raw = internal::SubtleMustCopy(image_size.flat<T>()(1));
argmax_op.cc 57 const int32 dim = internal::SubtleMustCopy(dimension.scalar<int32>()());
  /external/tensorflow/tensorflow/core/util/
strided_slice_op.cc 117 dense->begin[full_index] = internal::SubtleMustCopy<T>(begin_flat(i));
121 dense->end[full_index] = internal::SubtleMustCopy<T>(end_flat(i));
124 internal::SubtleMustCopy<T>(strides_flat(i));
  /external/tensorflow/tensorflow/contrib/factorization/kernels/
masked_matmul_ops.cc 120 int64 a_index = internal::SubtleMustCopy(indices_mat(i, 0));
127 int64 b_index = internal::SubtleMustCopy(indices_mat(i, 1));
  /external/tensorflow/tensorflow/contrib/tensor_forest/kernels/
tree_utils.h 185 int64 midi = internal::SubtleMustCopy(sparse_input_indices(mid, 0));
186 int64 midj = internal::SubtleMustCopy(sparse_input_indices(mid, 1));

Completed in 378 milliseconds

1 2