Home
last modified time | relevance | path

Searched defs:data_flat (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Ddynamic_partition_op.cc104 const auto data_flat = data->flat<T>(); in Compute() local
137 const auto data_flat = data->shaped<T, 2>({N, slice_size}); in Compute() local
Ddynamic_stitch_op.cc189 GpuDeviceArrayOnHost<const T*> data_flat(c, data_elements_size); in Compute() local
258 auto data_flat = in Compute() local
Dsegment_reduction_ops.cc383 auto data_flat = typename TTypes<T, 2>::ConstTensor(data, N, data_size / N); in operator ()() local
/external/tensorflow/tensorflow/cc/gradients/
Dimage_grad_test.cc46 auto data_flat = data.flat<T>(); in MakeData() local
191 auto data_flat = data.flat<T>(); in MakeData() local
308 auto data_flat = data.flat<T>(); in MakeData() local