Searched refs:tensor_eigen (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | summary_image_op.cc | 128 auto tensor_eigen = tensor.template shaped<T, 3>({batch_size, hw, depth}); in NormalizeAndAddImages() local
130 &tensor_eigen(i, 0, 0), in NormalizeAndAddImages()
|
| /external/tensorflow/tensorflow/core/summary/ |
| D | summary_converter.cc | 195 auto tensor_eigen = tensor.template shaped<T, 3>({batch_size, hw, depth}); in NormalizeAndAddImages() local
197 &tensor_eigen(i, 0, 0), Eigen::DSizes<Eigen::DenseIndex, 2>(hw, depth)); in NormalizeAndAddImages()
|