HomeSort by relevance Sort by last modified time
    Searched refs:data_flat (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
dynamic_stitch_op.cc 177 CudaDeviceArrayOnHost<const T*> data_flat(c, data_elements_size);
179 OP_REQUIRES_OK(c, data_flat.Init());
185 // data_flat index
197 data_flat.Set(
205 OP_REQUIRES_OK(c, data_flat.Finalize());
209 indices_flat.data(), data_flat.data(), output);
246 auto data_flat =
251 const T* data_base = &data_flat(0, 0);
271 data_flat.slice(data_indices, sizes);
dynamic_partition_op.cc 104 const auto data_flat = data->flat<T>(); variable
122 out_vec[p](oi) = data_flat(i);
137 const auto data_flat = data->shaped<T, 2>({N, slice_size}); variable
154 data_flat.slice(data_indices, sizes);
segment_reduction_ops.cc 379 auto data_flat = typename TTypes<T, 2>::ConstTensor(data, N, data_size / N); local
389 reduction(data_flat.template chip<0>(i), output.template chip<0>(j));
    [all...]

Completed in 119 milliseconds