Searched refs:then_handle (Results 1 – 2 of 2) sorted by relevance
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
D | select_op.cc | 45 auto then_handle = ctx->Input(1); in Compile() local 78 ctx->SetOutput(0, xla::Select(cond_handle, then_handle, else_handle)); in Compile()
|
/external/tensorflow/tensorflow/core/kernels/ |
D | functional_ops.cc | 150 FHandle then_handle; in ComputeAsync() local 152 OP_REQUIRES_OK_ASYNC(ctx, Instantiate(lib, then_func_, &then_handle), done); in ComputeAsync() 157 (new State(this, ctx, cond, then_handle, else_handle, done))->Start(); in ComputeAsync() 166 State(IfOp* kernel, OpKernelContext* ctx, bool cond, FHandle then_handle, in State() argument 171 then_handle_(then_handle), in State()
|