| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| select_test.cc | 41 auto on_false = builder.ConstantR0<float>(42.0f); local
42 auto result = builder.Select(pred, on_true, on_false);
51 auto on_false = builder.ConstantR0<int32>(42); local
52 auto result = builder.Select(pred, on_true, on_false);
61 auto on_false = builder.ConstantR0<float>(42.0f); local
62 auto result = builder.Select(pred, on_true, on_false);
71 auto on_false = builder.ConstantR1<float>({}); local
72 auto select = builder.Select(pred, on_true, on_false);
81 auto on_false = builder.ConstantR1<float>({10.0f, 5.0f, 1.0f, 10.0f, -6.0f}); local
82 auto select = builder.Select(pred, on_true, on_false);
96 auto on_false = builder.ConstantR1<float>({}); local
110 auto on_false = builder.ConstantR1<float>({10.0f, 5.0f, 1.0f, 10.0f, -6.0f}); local
124 auto on_false = builder.ConstantR1<float>({10.0f, 5.0f, 1.0f, 10.0f, -6.0f}); local
201 auto on_false = local
218 auto on_false = local
231 auto on_false = builder.ConstantR1<float>({}); local
242 auto on_false = builder.ConstantR1<float>({10.0f, 5.0f}); local
252 auto on_false = builder.ConstantR1<float>({10.0f, 5.0f}); local
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| tuple_ops.h | 63 llvm::Value* on_false, llvm::IRBuilder<>* ir_builder,
|
| tuple_ops.cc | 34 llvm::Value* on_false, llvm::IRBuilder<>* ir_builder,
58 ir_builder->CreateInBoundsGEP(on_false, element_index);
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| shape_inference.h | 281 const Shape& on_false);
|
| tuple_points_to_analysis.cc | 385 // on_false buffer into this new buffer. Which side is chosen cannot be
387 // of these on_true and on_false operands.
390 // add in elements of the on_false points-to set (tuple sources).
392 auto on_false = select->operand(2); local
394 const PointsToSet& false_points_to_set = *PerInst(on_false)->points_to_set;
|
| shape_inference.cc | [all...] |
| hlo_evaluator_test.cc | 160 auto on_false = Literal::CreateR2<float>({{0.f, 5.f}, {0.f, 4.f}}); local
168 b.AddInstruction(HloInstruction::CreateConstant(std::move(on_false)));
[all...] |
| hlo_evaluator.cc | 846 [](bool pred, ReturnT on_true, ReturnT on_false) {
850 return on_false;
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| xla_client.py | 743 def Select(self, pred, on_true, on_false):
753 _unwrap_data_handle(on_false)))
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| ir_emitter.cc | 424 auto on_false = select->operand(2); local
431 GetBasePointer(*on_false), &ir_builder_, module_);
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| computation_builder.cc | 471 const ComputationDataHandle& on_false) {
472 return TernaryOp(TRIOP_SELECT, pred, on_true, on_false);
[all...] |
| computation_builder.h | 299 // predicated on pred, selects between on_true and on_false.
302 const ComputationDataHandle& on_false);
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| ir_emitter.cc | 316 auto on_false = select->operand(2); local
323 GetEmittedValueFor(on_false), &ir_builder_, module_);
[all...] |
