Method Run

tensorflow/compiler/xla/service/depthwise_convolution_converter.cc:76–82

};

bool ConvolutionVisitor::Run(
    HloComputation* computation,
    std::function<bool(HloInstruction*)> is_cost_viable) {
  ConvolutionVisitor visitor(computation, is_cost_viable);
  TF_CHECK_OK(computation->Accept(&visitor));
  return visitor.changed_;
}

namespace {
Shape SwapInputOutputFeatureDims(const Shape& shape, int64 input_feature_dim,