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()
|