Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/tests/
Dselect_test.cc41 auto on_false = builder.ConstantR0<float>(42.0f); in TEST_F() local
51 auto on_false = builder.ConstantR0<int32>(42); in TEST_F() local
61 auto on_false = builder.ConstantR0<float>(42.0f); in TEST_F() local
71 auto on_false = builder.ConstantR1<float>({}); in XLA_TEST_F() local
81 auto on_false = builder.ConstantR1<float>({10.0f, 5.0f, 1.0f, 10.0f, -6.0f}); in TEST_F() local
96 auto on_false = builder.ConstantR1<float>({}); in XLA_TEST_F() local
110 auto on_false = builder.ConstantR1<float>({10.0f, 5.0f, 1.0f, 10.0f, -6.0f}); in TEST_F() local
124 auto on_false = builder.ConstantR1<float>({10.0f, 5.0f, 1.0f, 10.0f, -6.0f}); in TEST_F() local
201 auto on_false = in TEST_F() local
218 auto on_false = in TEST_F() local
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dtuple_ops.cc34 llvm::Value* on_false, llvm::IRBuilder<>* ir_builder, in EmitTupleSelect()
/external/tensorflow/tensorflow/compiler/xla/service/
Dtuple_points_to_analysis.cc392 auto on_false = select->operand(2); in HandleSelect() local
Dshape_inference.cc2383 const Shape& pred, const Shape& on_true, const Shape& on_false) { in InferSelectShape()
Dhlo_evaluator_test.cc160 auto on_false = Literal::CreateR2<float>({{0.f, 5.f}, {0.f, 4.f}}); in TEST_P() local
Dhlo_evaluator.cc846 [](bool pred, ReturnT on_true, ReturnT on_false) { in HandleSelect()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter.cc424 auto on_false = select->operand(2); in HandleSelect() local
/external/tensorflow/tensorflow/compiler/xla/python/
Dxla_client.py743 def Select(self, pred, on_true, on_false): argument
/external/tensorflow/tensorflow/compiler/xla/client/
Dcomputation_builder.cc471 const ComputationDataHandle& on_false) { in Select()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc316 auto on_false = select->operand(2); in HandleSelect() local