Searched refs:cond_value (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | resolve_constant_select.cc | 66 bool cond_value = cond_data[0]; in Run() local
68 if (cond_data[i] != cond_value) { in Run()
79 RemoveTrivialPassthroughOp(this, model, op_index, cond_value ? 1 : 2); in Run()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | legalize_control_flow.cc | 87 auto cond_value = builder.create<mlir::tensor::ExtractOp>(loc, if_op.pred()); in LowerIfOp() local
88 builder.create<mlir::CondBranchOp>(loc, cond_value, true_block, in LowerIfOp()
170 auto cond_value = in LowerWhileOp() local
176 builder.create<mlir::CondBranchOp>(loc, cond_value, body_block, in LowerWhileOp()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | optimize_functional_ops.cc | 110 bool cond_value = (*cond.int_value_begin()).getSExtValue(); in matchAndRewrite() local
111 FuncOp func = cond_value ? then_func : else_func; in matchAndRewrite()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | if.cc | 142 bool cond_value = cond->data.b[0]; in Eval() local
151 cond_value ? op_data->then_subgraph_index : op_data->else_subgraph_index; in Eval()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_alias_analysis.cc | 292 const HloValue& cond_value = dataflow_.GetUniqueValueAt( in ComputeConditionalAliasedBuffers() local
298 << cond_value.ToShortString(); in ComputeConditionalAliasedBuffers()
299 aliased_buffers->push_back(GetBufferForValue(cond_value)); in ComputeConditionalAliasedBuffers()
|