Home
last modified time | relevance | path

Searched defs:output_labels (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/util/
Deinsum_op_util.cc83 Labels* output_labels, in ParseEinsumEquation()
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_einsum_op.cc156 Labels output_labels(mkl_output_labels_); in Compute() local
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ops/
Deinsum.cc565 Labels output_labels) { in GetOutputTranspose()
603 Labels output_labels; in ParseEquation() local
/external/tensorflow/tensorflow/python/keras/engine/
Dtraining_distributed_v1.py55 def _make_train_step_fn(model, mode, strategy, output_labels): argument
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.cc162 string output_labels; in EinsumShape() local