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

  /external/tensorflow/tensorflow/core/kernels/
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 138 milliseconds