Searched refs:pred_value (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/python/framework/ |
| D | smart_cond.py | 51 pred_value = smart_constant_value(pred)
52 if pred_value is not None:
53 if pred_value:
75 pred_value = bool(pred)
77 pred_value = pred
79 pred_value = tensor_util.constant_value(pred)
82 if pred_value is None:
83 pred_value = c_api.TF_TryEvaluateConstant_wrapper(pred.graph._c_graph,
90 return pred_value
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | tuple_ops.cc | 42 llvm::LoadInst* pred_value = in EmitTupleSelect() local
45 pred_value, in EmitTupleSelect()
50 VLOG(2) << " pred_value: " << DumpToString(*pred_value); in EmitTupleSelect()
|
| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| D | utils.py | 211 pred_value = constant_value(pred)
212 if pred_value is not None:
214 return static_cond(pred_value, fn1, fn2)
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | cond_v2_test.py | 283 def run_test(pred_value): argument
307 self._testCond(true_fn, false_fn, [x, y], {pred: pred_value})
308 self._testCond(true_fn, false_fn, [x], {pred: pred_value})
309 self._testCond(true_fn, false_fn, [y], {pred: pred_value})
316 def run_test(pred_value): argument
333 self._testCond(true_fn, false_fn, [x, y], {pred: pred_value})
334 self._testCond(true_fn, false_fn, [x], {pred: pred_value})
335 self._testCond(true_fn, false_fn, [y], {pred: pred_value})
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | elemental_ir_emitter.cc | 1672 TF_ASSIGN_OR_RETURN(llvm::Value * pred_value, in EmitElementalSelect()
1678 return Select(Trunc(pred_value, b_->getInt1Ty()), on_true_value, in EmitElementalSelect()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 2558 llvm::LoadInst* pred_value = Load( in HandleConditional() local
2561 ICmpNE(pred_value, in HandleConditional()
|