Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 6 of 6 for DataDumperLoggerConfig (0.47 sec)

  2. github.com/tensorflow/tensorflow

    tensorflow/compiler/mlir/tensorflow/utils/data_dumper_logger_config.cc 

    #include <functional>
#include <memory>
#include <string>

#include "tensorflow/compiler/mlir/tensorflow/utils/dump_mlir_util.h"

namespace tensorflow {
DataDumperLoggerConfig::DataDumperLoggerConfig(
    std::function<std::string(const std::string &, mlir::Operation *op)>
        get_filename,
    const std::string &pass_prefix, bool print_module_scope,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Aug 31 00:41:24 UTC 2023
    - 2.4K bytes
    - Viewed (0)
  3. github.com/tensorflow/tensorflow

    tensorflow/compiler/mlir/tensorflow/utils/data_dumper_logger_config.h 

    #include <functional>
#include <string>

#include "tensorflow/compiler/mlir/tensorflow/utils/bridge_logger.h"

namespace tensorflow {

class DataDumperLoggerConfig : public ::tensorflow::BridgeLoggerConfig {
 public:
  explicit DataDumperLoggerConfig(
      std::function<std::string(const std::string &, mlir::Operation *op)>
          get_filename,
      const std::string &pass_prefix = "", bool print_module_scope = false,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Aug 31 00:41:24 UTC 2023
    - 2.1K bytes
    - Viewed (0)
  4. github.com/tensorflow/tensorflow

    tensorflow/compiler/mlir/tensorflow/utils/data_dumper_logger_config_test.cc 

      %0 = "tf.AddV2"(%arg0, %arg1) : (tensor<3x4x5xf32>, tensor<3x4x5xf32>) -> tensor<3x4x5xf32>
  func.return %0 : tensor<3x4x5xf32>
}
}
)";

// Test pass filter.
TEST(DataDumperLoggerConfig, TestPassFilter) {
  mlir::DialectRegistry mlir_registry;
  mlir::RegisterAllTensorFlowDialects(mlir_registry);
  mlir::MLIRContext mlir_context(mlir_registry);
  mlir::OwningOpRef<mlir::ModuleOp> mlir_module_with_add;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Aug 31 00:41:24 UTC 2023
    - 3.9K bytes
    - Viewed (0)
  5. github.com/tensorflow/tensorflow

    tensorflow/compiler/mlir/tf2xla/internal/logging_hooks.cc 

      // Print the whole module after each pass, which requires disabling
  // multi-threading as well.
  pm.getContext()->disableMultithreading();
  pm.enableIRPrinting(std::make_unique<::tensorflow::DataDumperLoggerConfig>(
      [module_name, dump_group_name](const std::string& pass_tag_name,
                                     mlir::Operation* op) {
        return DEBUG_DATA_DUMPER()->GetDumpFilename(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Oct 05 14:26:22 UTC 2023
    - 2K bytes
    - Viewed (0)
  6. github.com/tensorflow/tensorflow

    tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/lower_cluster_to_runtime_ops.cc 

      // Print the whole module after each pass, which requires disabling
  // multi-threading as well.
  pm.getContext()->disableMultithreading();
  pm.enableIRPrinting(std::make_unique<::tensorflow::DataDumperLoggerConfig>(
      [module_name, dump_group_name](const std::string& pass_tag_name,
                                     mlir::Operation* op) {
        return DEBUG_DATA_DUMPER()->GetDumpFilename(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Apr 17 18:52:57 UTC 2024
    - 9.4K bytes
    - Viewed (0)
  7. github.com/tensorflow/tensorflow

    tensorflow/compiler/mlir/tf2xla/api/v1/compile_mlir_util.cc 

        // multi-threading as well.
    module_op.getContext()->disableMultithreading();
    tf2xla.enableIRPrinting(
        std::make_unique<::tensorflow::DataDumperLoggerConfig>(
            [module_name](const std::string& pass_tag_name,
                          mlir::Operation* op) {
              return DEBUG_DATA_DUMPER()->GetDumpFilename(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 21 17:24:39 UTC 2024
    - 45.3K bytes
    - Viewed (0)
Back to top