Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for GraphdefToSplattedMlirTranslateFunction (0.32 sec)

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

    }
    
    static TranslateToMLIRRegistration GraphdefToMlirTranslate(
        "graphdef-to-mlir", "graphdef-to-mlir", GraphdefToMlirTranslateFunction);
    
    static OwningOpRef<mlir::ModuleOp> GraphdefToSplattedMlirTranslateFunction(
        llvm::StringRef input, MLIRContext* context) {
      tensorflow::GraphdefToMlirOptions options{
          debug_info_file,        xla_compile_device_type,
          prune_unused_nodes,     convert_legacy_fed_inputs,
    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/tensorflow/translate/tf_mlir_translate.h

    // Similar as the above function, but replaces all constant tensors
    // with randomly generated splat values.
    absl::StatusOr<mlir::OwningOpRef<mlir::ModuleOp>>
    GraphdefToSplattedMlirTranslateFunction(
        llvm::StringRef input, const std::vector<std::string>& input_arrays,
        const std::vector<std::string>& input_dtypes,
        const std::vector<std::vector<int>>& input_shapes,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 01 11:17:36 UTC 2024
    - 5.9K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/translate/tf_mlir_translate.cc

      if (!module_or.status().ok()) {
        LOG(ERROR) << "SavedModel V1 import failed: " << module_or.status();
      }
      return module_or;
    }
    
    absl::StatusOr<mlir::OwningOpRef<mlir::ModuleOp>>
    GraphdefToSplattedMlirTranslateFunction(
        llvm::StringRef input, const std::vector<std::string>& input_arrays,
        const std::vector<std::string>& input_dtypes,
        const std::vector<std::optional<std::vector<int>>>& input_shapes,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 07 11:51:44 UTC 2024
    - 14.1K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/lite/tf_to_tfl_flatbuffer.cc

          /*enable_shape_inference=*/false,
          /*unconditionally_use_set_output_shapes=*/true,
          /*enable_soft_placement=*/false};
    
      if (use_splatted_constant) {
        return GraphdefToSplattedMlirTranslateFunction(
            file->getBuffer(), input_arrays, input_dtypes, input_shapes,
            output_arrays, control_output_arrays, graphdef_conversion_options,
            context);
      }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 03 18:01:23 UTC 2024
    - 23.8K bytes
    - Viewed (0)
Back to top