Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for enable_soft_placement (0.15 sec)

  1. tensorflow/compiler/mlir/tensorflow/translate/tf_mlir_translate_registration.cc

          prune_unused_nodes,     convert_legacy_fed_inputs,
          graph_as_function,      upgrade_legacy,
          enable_shape_inference, unconditionally_use_set_output_shapes,
          enable_soft_placement,  set_original_tf_func_name};
    
      auto module_or = tensorflow::GraphdefToMlirTranslateFunction(
          input, input_arrays, input_dtypes, input_shapes, output_arrays,
          control_output_arrays, options, context);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 22:19:26 UTC 2024
    - 7.8K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/mlir_graph_optimization_pass.cc

      // during import is not necessary.
      import_config.enable_shape_inference = false;
      import_config.xla_compile_device_type =
          function_options.xla_compile_device_type;
      import_config.enable_soft_placement = function_options.allow_soft_placement;
    
      static const char* kTfMlirCategory = "TfMlir";
      tensorflow::metrics::ScopedCounter<2> timings(
          tensorflow::metrics::GetGraphOptimizationCounter(),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 22:19:26 UTC 2024
    - 18.5K bytes
    - Viewed (0)
Back to top