Home
last modified time | relevance | path

Searched defs:cond_output (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dwhile.cc191 const TfLiteTensor* cond_output) { in CheckCondOutput()
257 TfLiteTensor* cond_output = in Prepare_impl() local
365 TfLiteTensor* cond_output = cond_subgraph->tensor(cond_subgraph_output_index); in Eval_cond_subgraph() local
/aosp_15_r20/external/tensorflow/tensorflow/core/graph/
H A Dwhile_context.cc23 OutputTensor cond_output, in WhileContext()
H A Dwhile_context.h45 const OutputTensor& cond_output() const { return cond_output_; } in cond_output() function
H A Dgraph.cc920 OutputTensor cond_output, in AddWhileContext()
/aosp_15_r20/external/tensorflow/tensorflow/c/
H A Dc_api.h933 TF_Output cond_output; member
H A Dc_api.cc2027 std::vector<tensorflow::Output> cond_output; in TF_FinishWhileHelper() local
/aosp_15_r20/external/tensorflow/tensorflow/cc/ops/
H A Dwhile_loop.cc176 Output* cond_output) { in BuildWhileLoop()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/
H A Dfunctionalize_while.cc121 std::unique_ptr<Graph>* cond_output) { in BuildLoopCondition()
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Dloop_unrolling.cpp254 auto cond_output = loop->addOutput(); in addCondAsOutput() local