Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dcase_op.cc327 xla::XlaOp token_output = in Compile() local
H A Dif_op.cc345 xla::XlaOp token_output = in Compile() local
H A Dwhile_op.cc607 xla::XlaOp token_output = in Compile() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/
H A Dxla_compiler.cc172 std::unique_ptr<xla::XlaOp> token_output, in BuildComputation()
1346 std::unique_ptr<xla::XlaOp> token_output; in CompileGraph() local
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/xla/
H A Dhost_compute_ops.cc251 xla::XlaOp token_output = recv_from_host_tokens.empty() in Compile() local