Searched defs:on_false (Results 1 – 11 of 11) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/tests/ |
D | select_test.cc | 40 auto on_false = ConstantR0<float>(&builder, 42.0f); in TEST_F() local 50 auto on_false = ConstantR0<int32>(&builder, 42); in TEST_F() local 60 auto on_false = ConstantR0<float>(&builder, 42.0f); in TEST_F() local 70 auto on_false = ConstantR1<float>(&builder, {}); in XLA_TEST_F() local 81 auto on_false = in TEST_F() local 97 auto on_false = ConstantR1<float>(&builder, {}); in XLA_TEST_F() local 112 auto on_false = in TEST_F() local 128 auto on_false = in TEST_F() local 206 auto on_false = in TEST_F() local 223 auto on_false = in TEST_F() local [all …]
|
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
D | tuple_ops.cc | 37 llvm::Value* on_true, llvm::Value* on_false, in EmitTupleSelect()
|
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | hlo_creation_utils.cc | 277 HloInstruction* on_false) { in MakeSelectHlo()
|
D | shape_inference.cc | 2773 const Shape& pred, const Shape& on_true, const Shape& on_false) { in InferSelectShape() 2812 const Shape& pred, const Shape& on_true, const Shape& on_false) { in InferTupleSelectShape()
|
D | hlo_evaluator.cc | 1291 const auto& on_false = GetEvaluatedLiteralFor(select->operand(2)); in HandleSelect() local 1309 const auto& on_false = GetEvaluatedLiteralFor(tuple_select->operand(2)); in HandleTupleSelect() local
|
D | tuple_points_to_analysis.cc | 404 auto on_false = tuple_select->operand(2); in HandleTupleSelect() local
|
D | hlo_evaluator_test.cc | 226 auto on_false = LiteralUtil::CreateR2<float>({{0.f, 5.f}, {0.f, 4.f}}); in TEST_P() local
|
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | ir_emitter.cc | 430 auto on_false = tuple_select->operand(2); in HandleTupleSelect() local
|
/external/tensorflow/tensorflow/compiler/xla/client/ |
D | xla_builder.cc | 965 const XlaOp& on_false) { in Select() 2909 XlaOp Select(const XlaOp& pred, const XlaOp& on_true, const XlaOp& on_false) { in Select()
|
/external/tensorflow/tensorflow/compiler/xla/python/ |
D | xla_client.py | 1309 def Select(self, pred, on_true, on_false): argument
|
/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
D | ir_emitter.cc | 314 auto on_false = tuple_select->operand(2); in HandleTupleSelect() local
|