Searched refs:kClassDim (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | softmax_op_functor.h | 47 const int kClassDim = 1; in Compute() local
50 const int num_classes = logits.dimension(kClassDim); in Compute()
55 Eigen::DSizes<int, 1> along_class(kClassDim); in Compute()
59 Eigen::IndexList<Eigen::type2index<kClassDim> > along_class; in Compute()
|
| D | xent_op.h | 67 const int kClassDim = 1; in Compute() local
70 const int num_classes = shape[kClassDim]; in Compute()
76 along_class[0] = kClassDim; in Compute()
86 Eigen::IndexList<Eigen::type2index<kClassDim> > along_class; in Compute()
|
| D | sparse_xent_op.h | 184 const int kClassDim = 1; in Compute() local
187 const int num_classes = logits.dimension(kClassDim); in Compute()
193 along_class[0] = kClassDim; in Compute()
203 Eigen::IndexList<Eigen::type2index<kClassDim> > along_class; in Compute()
|
| D | sparse_xent_op_gpu.cu.cc | 45 const int kClassDim = 1; in Compute() local
47 const int cols = logits.dimension(kClassDim); in Compute()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | softmax_op.cc | 49 const int kClassDim = logits_shape.dims() - 1; in Compile() local
60 xla::Reduce(logits, xla::MinValue(b, xla_type), max_func, {kClassDim}); in Compile()
73 *ctx->GetOrCreateAdd(accumulation_type), {kClassDim}); in Compile()
97 const int kClassDim = 1; in CrossEntropyWithLogits() local
102 xla::Reduce(logits, xla::MinValue(b, xla_type), max_func, {kClassDim}); in CrossEntropyWithLogits()
117 *ctx->GetOrCreateAdd(accumulation_type), {kClassDim}); in CrossEntropyWithLogits()
131 *ctx->GetOrCreateAdd(accumulation_type), {kClassDim}); in CrossEntropyWithLogits()
|