Searched refs:add_computation (Results 1 – 3 of 3) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | hlo_evaluator_test.cc | 2527 HloComputation::Builder add_computation("add"); in TEST_F() local 2529 auto param_lhs = add_computation.AddInstruction( in TEST_F() 2531 auto param_rhs = add_computation.AddInstruction( in TEST_F() 2533 add_computation.AddInstruction(HloInstruction::CreateBinary( in TEST_F() 2535 auto add_func = m->AddEmbeddedComputation(add_computation.Build()); in TEST_F() 2562 HloComputation::Builder add_computation("add"); in BM_ReducePrecisely() local 2564 auto param_lhs = add_computation.AddInstruction( in BM_ReducePrecisely() 2566 auto param_rhs = add_computation.AddInstruction( in BM_ReducePrecisely() 2568 add_computation.AddInstruction(HloInstruction::CreateBinary( in BM_ReducePrecisely() 2570 auto add_func = module.AddEmbeddedComputation(add_computation.Build()); in BM_ReducePrecisely() [all …]
|
D | algebraic_simplifier_test.cc | 590 HloComputation* add_computation = nullptr; in TEST_F() local 600 add_computation = m->AddEmbeddedComputation(builder.Build()); in TEST_F() 608 HloInstruction::CreateReduce(r3f32, param, zero, dims0, add_computation)); in TEST_F() 612 dims1, add_computation)); in TEST_F() 717 HloComputation* add_computation = nullptr; in TEST_F() local 727 add_computation = m->AddEmbeddedComputation(builder.Build()); in TEST_F() 738 add_computation)); in TEST_F() 1765 HloComputation* add_computation = nullptr; in TEST_F() local 1775 add_computation = m->AddEmbeddedComputation(builder.Build()); in TEST_F() 1781 window, add_computation)); in TEST_F() [all …]
|
/external/tensorflow/tensorflow/compiler/xla/client/lib/ |
D | pooling.cc | 82 auto add_computation = CreateScalarAddComputation(accumulation_type, b); in ComputeSums() local 83 return ReduceWindow(operand, init_value, add_computation, kernel_size, in ComputeSums()
|