Searched refs:cond_handle (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | select_op.cc | 44 auto cond_handle = ctx->Input(0); in Compile() local
71 cond_handle = xla::Broadcast(cond_handle, bdims); in Compile()
76 cond_handle = xla::Transpose(cond_handle, dim_order); in Compile()
78 ctx->SetOutput(0, xla::Select(cond_handle, then_handle, else_handle)); in Compile()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | const_analysis.cc | 142 FunctionLibraryRuntime::Handle cond_handle; in GetCompileTimeConstInputs() local
145 cond_function.name(), AttrSlice(&cond_function.attr()), &cond_handle)); in GetCompileTimeConstInputs()
148 const FunctionBody* fcond = flib_runtime->GetFunctionBody(cond_handle); in GetCompileTimeConstInputs()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | functional_ops.cc | 344 FHandle cond_handle; in ComputeAsync() local
346 OP_REQUIRES_OK_ASYNC(ctx, Instantiate(lib, cond_func_, &cond_handle), done); in ComputeAsync()
348 (new State(this, ctx, cond_handle, body_handle, done))->Start(); in ComputeAsync()
357 State(WhileOp* kernel, OpKernelContext* ctx, FHandle cond_handle, in State() argument
361 cond_handle_(cond_handle), in State()
|