HomeSort by relevance Sort by last modified time
    Searched defs:add_computation (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/client/lib/
pooling.cc 82 auto add_computation = CreateScalarAddComputation(accumulation_type, b); local
83 return ReduceWindow(operand, init_value, add_computation, kernel_size,
  /external/tensorflow/tensorflow/compiler/xla/service/
algebraic_simplifier_test.cc 369 HloComputation* add_computation = nullptr; local
379 add_computation = m->AddEmbeddedComputation(builder.Build());
387 HloInstruction::CreateReduce(r3f32, param, zero, dims0, add_computation));
391 dims1, add_computation));
475 HloComputation* add_computation = nullptr; local
485 add_computation = m->AddEmbeddedComputation(builder.Build());
496 add_computation));
1401 HloComputation* add_computation = nullptr; local
1444 HloComputation* add_computation = nullptr; local
1724 HloComputation* add_computation = nullptr; local
3592 HloComputation* add_computation = nullptr; local
3678 HloComputation* add_computation = nullptr; local
4244 HloComputation* add_computation = nullptr; local
    [all...]

Completed in 97 milliseconds