Searched refs:Tensor32Bit (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_xent_op.h | 32 typename TTypes<const T, 1>::Tensor32Bit To32BitConst( in To32BitConst()
38 typename TTypes<const T, 2>::Tensor32Bit To32BitConst( in To32BitConst()
60 typename TTypes<const T, 2>::Tensor32Bit logits, in SparseXentLossGenerator()
61 typename TTypes<const T, 1>::Tensor32Bit sum_exp_logits, in SparseXentLossGenerator()
62 typename TTypes<const Index, 1>::Tensor32Bit labels, in SparseXentLossGenerator()
83 typename TTypes<const T, 2>::Tensor32Bit logits_;
84 typename TTypes<const T, 1>::Tensor32Bit sum_exp_logits_;
85 typename TTypes<const Index, 1>::Tensor32Bit labels_;
100 typename TTypes<const T, 2>::Tensor32Bit exp_logits, in SparseXentGradGenerator()
101 typename TTypes<const T, 1>::Tensor32Bit sum_exp_logits, in SparseXentGradGenerator()
[all …]
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor_types.h | 43 Tensor32Bit; typedef
105 typedef typename TTypes<float, 1>::Tensor32Bit::Index Index32;
118 TensorType::NumIndices>::Tensor32Bit
121 TensorType::NumIndices>::Tensor32Bit RetType; in To32Bit()
|