Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/examples/custom_ops_doc/multiplex_3/
H A Dmultiplex_3_kernel.cc142 Tensor* output_indices_tensor; in Compute() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
H A Dconvert_nodes.cc4831 ITensorProxyPtr output_indices_tensor = layer->getOutput(1); in ConvertArgMinMax() local
4889 ITensorProxyPtr output_indices_tensor = layer->getOutput(1); in ConvertTopK() local