Home
last modified time | relevance | path

Searched refs:add_computation (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator_test.cc2527 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 …]
Dalgebraic_simplifier_test.cc590 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/
Dpooling.cc82 auto add_computation = CreateScalarAddComputation(accumulation_type, b); in ComputeSums() local
83 return ReduceWindow(operand, init_value, add_computation, kernel_size, in ComputeSums()