HomeSort by relevance Sort by last modified time
    Searched full:reduce_window (Results 1 - 11 of 11) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/
dfs_hlo_visitor_with_default.h 152 Status HandleReduceWindow(HloInstructionPtr reduce_window) override {
153 return DefaultAction(reduce_window);
algebraic_simplifier.cc 161 Status HandleReduceWindow(HloInstruction* reduce_window) override;
    [all...]
hlo_verifier.cc 263 Status ShapeVerifier::HandleReduceWindow(HloInstruction* reduce_window) {
265 reduce_window,
267 reduce_window->operand(0)->shape(),
268 reduce_window->operand(1)->shape(), reduce_window->window(),
269 reduce_window->to_apply()->ComputeProgramShape()));
    [all...]
hlo_cost_analysis.cc 270 const HloInstruction* reduce_window) {
271 const Window& window = reduce_window->window();
272 auto function = reduce_window->to_apply();
284 ShapeUtil::ElementsIn(reduce_window->shape());
hlo_verifier.h 70 Status HandleReduceWindow(HloInstruction* reduce_window) override;
hlo_cost_analysis.h 94 Status HandleReduceWindow(const HloInstruction* reduce_window) override;
hlo_evaluator.cc     [all...]
hlo_cost_analysis_test.cc 232 ComputationBuilder builder(client_, "reduce_window");
algebraic_simplifier_test.cc 2276 HloInstruction* reduce_window = local
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
ir_emitter.cc 514 Status IrEmitter::HandleReduceWindow(HloInstruction* reduce_window) {
515 auto operand = reduce_window->operand(0);
516 const Window& window = reduce_window->window();
517 HloComputation* function = reduce_window->to_apply();
519 /*instruction=*/*reduce_window, /*operands=*/{operand},
545 reduce_window, [this, reduce_window, operand, window,
548 // the initial value on the reduce_window.
555 reduce_window->operand(1))),
558 llvm_ir::ForLoopNest loops(IrName(reduce_window, "inner")
    [all...]
ir_emitter.h 134 Status HandleReduceWindow(HloInstruction* reduce_window) override;

Completed in 702 milliseconds