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.cc324 HloInstruction* on_false, in MakeSelectHlo()
Dtuple_points_to_analysis.cc452 auto on_false = tuple_select->operand(2); in HandleTupleSelect() local
Dshape_inference.cc2984 const Shape& pred, const Shape& on_true, const Shape& on_false) { in InferSelectShape()
3014 const Shape& pred, const Shape& on_true, const Shape& on_false) { in InferTupleSelectShape()
Dhlo_evaluator.cc1991 const auto& on_false = GetEvaluatedLiteralFor(select->operand(2)); in HandleSelect() local
2009 const auto& on_false = GetEvaluatedLiteralFor(tuple_select->operand(2)); in HandleTupleSelect() local
Dhlo_evaluator_test.cc295 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.cc467 auto on_false = tuple_select->operand(2); in HandleTupleSelect() local
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc1084 XlaOp XlaBuilder::Select(XlaOp pred, XlaOp on_true, XlaOp on_false) { in Select()
2974 XlaOp Select(const XlaOp pred, const XlaOp on_true, const XlaOp on_false) { in Select()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc373 auto on_false = tuple_select->operand(2); in HandleTupleSelect() local
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc1343 Value on_false = broadcast_or_self(op.e()); in matchAndRewrite() local