Searched refs:cond_handle (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | select_op.cc | 46 auto cond_handle = ctx->Input(0); in Compile() local
70 cond_handle = xla::BroadcastInDim(cond_handle, then_shape.dim_sizes(), in Compile()
73 ctx->SetOutput(0, xla::Select(cond_handle, then_handle, else_handle)); in Compile()
116 auto cond_handle = bcasted_cond.ValueOrDie(); in Compile() local
126 ctx->SetOutput(0, xla::Select(cond_handle, then_handle, else_handle)); in Compile()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | functional_ops.cc | 371 FHandle cond_handle; in ComputeAsync() local
373 OP_REQUIRES_OK_ASYNC(ctx, GetHandles(ctx, &cond_handle, &body_handle), in ComputeAsync()
375 (new State(this, ctx, cond_handle, body_handle, done))->Start(); in ComputeAsync()
503 State(WhileOp* kernel, OpKernelContext* ctx, FHandle cond_handle, in State() argument
507 cond_handle_(cond_handle), in State()
603 FHandle cond_handle; in DoComputeSync() local
605 TF_RETURN_IF_ERROR(GetHandles(ctx, &cond_handle, &body_handle)); in DoComputeSync()
627 TF_RETURN_IF_ERROR(lib->RunSync(opts, cond_handle, args, &cond_rets)); in DoComputeSync()
657 Status GetHandles(OpKernelContext* ctx, FHandle* cond_handle, in GetHandles() argument
668 *cond_handle = kInvalidHandle; in GetHandles()
[all …]
|