Home
last modified time | relevance | path

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

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