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.cc1441 HloComputation::Builder add_computation("add"); in TEST_F() local
1443 auto param_lhs = add_computation.AddInstruction( in TEST_F()
1445 auto param_rhs = add_computation.AddInstruction( in TEST_F()
1447 add_computation.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
1449 auto add_func = m->AddEmbeddedComputation(add_computation.Build()); in TEST_F()
1477 HloComputation::Builder add_computation("add"); in BM_ReducePrecisely() local
1479 auto param_lhs = add_computation.AddInstruction( in BM_ReducePrecisely()
1481 auto param_rhs = add_computation.AddInstruction( in BM_ReducePrecisely()
1483 add_computation.AddInstruction(HloInstruction::CreateBinary( in BM_ReducePrecisely()
1485 auto add_func = module.AddEmbeddedComputation(add_computation.Build()); in BM_ReducePrecisely()
[all …]
Dalgebraic_simplifier_test.cc369 HloComputation* add_computation = nullptr; in TEST_F() local
379 add_computation = m->AddEmbeddedComputation(builder.Build()); in TEST_F()
387 HloInstruction::CreateReduce(r3f32, param, zero, dims0, add_computation)); in TEST_F()
391 dims1, add_computation)); in TEST_F()
475 HloComputation* add_computation = nullptr; in TEST_F() local
485 add_computation = m->AddEmbeddedComputation(builder.Build()); in TEST_F()
496 add_computation)); in TEST_F()
1401 HloComputation* add_computation = nullptr; in TEST_F() local
1411 add_computation = m->AddEmbeddedComputation(builder.Build()); in TEST_F()
1417 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()