Home
last modified time | relevance | path

Searched refs:Tensor32Bit (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dsparse_xent_op.h32 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/
Dtensor_types.h43 Tensor32Bit; typedef
105 typedef typename TTypes<float, 1>::Tensor32Bit::Index Index32;
118 TensorType::NumIndices>::Tensor32Bit
121 TensorType::NumIndices>::Tensor32Bit RetType; in To32Bit()