Searched refs:ApproxTopKFn (Results 1 – 1 of 1) sorted by relevance
88 xla::XlaOp output_tuple = ApproxTopKFn( in Compile()97 virtual xla::XlaOp ApproxTopKFn(120 xla::XlaOp ApproxTopKFn( in ApproxTopKFn() function in tensorflow::__anondd4c53d00111::TpuApproxTopKOp138 xla::XlaOp ApproxTopKFn( in ApproxTopKFn() function in tensorflow::__anondd4c53d00111::FallbackApproxTopKOp