Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/cc/gradients/
H A 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
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A 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
H A Ddynamic_stitch_op.cc190 GpuDeviceArrayOnHost<const T*> data_flat(c, data_elements_size); in Compute() local
260 auto data_flat = in Compute() local