Home
last modified time | relevance | path

Searched defs:labels_shape (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dsoftmax_op.cc149 const TensorShape labels_shape = ctx->InputShape(1); in Compile() local
182 const TensorShape labels_shape = ctx->InputShape(1); in Compile() local
/external/tensorflow/tensorflow/core/kernels/
Dctc_loss_op.cc101 TensorShape labels_shape({batch_size, max_label_len}); in Compute() local