H A D | nn_xent_test.py | 46 def _Inputs(self, x=None, y=None, dtype=dtypes.float64, sizes=None): member in class:SigmoidCrossEntropyWithLogitsTest 58 logits, targets, _ = self._Inputs() 67 logits, targets, losses = self._Inputs(dtype=dtype) 78 logits, targets, losses = self._Inputs(dtype=dtype, sizes=[2, 2, 2]) 88 logits, targets, _ = self._Inputs(sizes=sizes) 122 def _Inputs(self, x=None, y=None, q=3.0, dtype=dtypes.float64, sizes=None): member in class:WeightedCrossEntropyTest 134 logits, targets, pos_weight, _ = self._Inputs() 142 logits, targets, pos_weight, losses = self._Inputs(dtype=dtypes.float32) 152 logits, targets, pos_weight, losses = self._Inputs( 163 logits, targets, pos_weight, _ = self._Inputs(size [all...] |