Home
last modified time | relevance | path

Searched defs:on_false (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dselect_test.cc40 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/
Dtuple_ops.cc37 llvm::Value* on_true, llvm::Value* on_false, in EmitTupleSelect()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_creation_utils.cc277 HloInstruction* on_false) { in MakeSelectHlo()
Dshape_inference.cc2773 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()
Dhlo_evaluator.cc1291 const auto& on_false = GetEvaluatedLiteralFor(select->operand(2)); in HandleSelect() local
1309 const auto& on_false = GetEvaluatedLiteralFor(tuple_select->operand(2)); in HandleTupleSelect() local
Dtuple_points_to_analysis.cc404 auto on_false = tuple_select->operand(2); in HandleTupleSelect() local
Dhlo_evaluator_test.cc226 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/
Dir_emitter.cc430 auto on_false = tuple_select->operand(2); in HandleTupleSelect() local
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc965 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/
Dxla_client.py1309 def Select(self, pred, on_true, on_false): argument
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc314 auto on_false = tuple_select->operand(2); in HandleTupleSelect() local