Searched refs:data_flat (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | image_grad_test.cc | 46 auto data_flat = data.flat<T>(); in MakeData() local
47 for (int i = 0; i < data_flat.size(); ++i) { in MakeData()
48 data_flat(i) = T(i); in MakeData()
191 auto data_flat = data.flat<T>(); in MakeData() local
192 for (int i = 0; i < data_flat.size(); ++i) { in MakeData()
193 data_flat(i) = T(i); in MakeData()
308 auto data_flat = data.flat<T>(); in MakeData() local
309 for (int i = 0; i < data_flat.size(); ++i) { in MakeData()
310 data_flat(i) = T(i); in MakeData()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | dynamic_stitch_op.cc | 189 GpuDeviceArrayOnHost<const T*> data_flat(c, data_elements_size); in Compute() local
191 OP_REQUIRES_OK(c, data_flat.Init()); in Compute()
209 data_flat.Set( in Compute()
217 OP_REQUIRES_OK(c, data_flat.Finalize()); in Compute()
221 indices_flat.data(), data_flat.data(), output); in Compute()
258 auto data_flat = in Compute() local
263 const T* data_base = data_flat.data(); in Compute()
283 data_flat.slice(data_indices, sizes); in Compute()
|
| D | dynamic_partition_op.cc | 104 const auto data_flat = data->flat<T>(); in Compute() local
122 out_vec[p](oi) = data_flat(i); in Compute()
137 const auto data_flat = data->shaped<T, 2>({N, slice_size}); in Compute() local
154 data_flat.slice(data_indices, sizes); in Compute()
|
| D | segment_reduction_ops.cc | 383 auto data_flat = typename TTypes<T, 2>::ConstTensor(data, N, data_size / N); in operator ()() local
393 reduction(data_flat.template chip<0>(i), output.template chip<0>(j)); in operator ()()
|