Search Options

Results per page
Sort
Preferred Languages
Advance

Results 71 - 80 of 90 for input_type (0.12 sec)

  1. tensorflow/compiler/mlir/tensorflow/transforms/cluster_tf_ops_pass.cc

        // Do not create any new function for the operations on the localhost.
        if (IsOnLocalHost(host)) continue;
    
        llvm::SmallVector<mlir::Type, 4> input_types;
        llvm::SmallVector<mlir::Type, 4> result_types;
        for (Value input : metadata.inputs) {
          input_types.push_back(input.getType());
        }
        for (Value result : metadata.results) {
          result_types.push_back(result.getType());
        }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 13.4K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/experimental/common/outline_operations.cc

      // within the subgraph and are referenced outside the subgraph.
      llvm::SmallVector<Type> input_types =
          TypesFromValues(subgraph.FuncArguments());
      llvm::SmallVector<Type> return_types =
          TypesFromValues(subgraph.FuncOutputs());
    
      FunctionType function_type =
          builder.getFunctionType(input_types, return_types);
    
      std::string function_name = absl::StrCat("func_", subgraph.subgraph_id_);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 8.5K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/translate/import_model.cc

        auto* context = shape_refiner_->GetContext(&node);
        DataType dtype = node.input_type(idx);
        return ConvertDataTypeAndShape(dtype, context->input(idx),
                                       context->input_handle_shapes_and_types(idx),
                                       context, builder);
      }
      DataType dtype = node.properties()->input_types[idx];
      mlir::Type element_type;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 01 11:17:36 UTC 2024
    - 183.2K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tfr/integration/tfr_decompose_ctx.cc

      mlir::MLIRContext* context = tfr_module_.getContext();
      llvm::SmallVector<mlir::Type, 4> input_tys, output_tys;
      mlir::Builder builder(context);
      for (auto ty : input_dtys) {
        mlir::Type elt_ty;
        TF_RETURN_IF_ERROR(ConvertDataType(ty, builder, &elt_ty));
        mlir::TensorType mlir_ty = mlir::UnrankedTensorType::get(elt_ty);
        input_tys.push_back(mlir_ty);
      }
      for (auto ty : output_dtys) {
        mlir::Type elt_ty;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 29 02:34:43 UTC 2024
    - 9.1K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/quantization/tensorflow/passes/merge_initializer_function_ops_to_main.cc

    // attaching a to the initializer function's type so that it is identifiable.
    Location CreateInitOpLoc(MLIRContext* ctx, func::FuncOp init_func_ops) {
      const std::string init_type = GetInitializerType(init_func_ops);
      const std::string name =
          absl::StrCat(init_type, "_", init_func_ops.getName().str());
      return NameLoc::get(StringAttr::get(ctx, name));
    }
    
    void MergeInitializerFunctionOpsToMainPass::runOnOperation() {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sun May 12 12:54:52 UTC 2024
    - 15.3K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tensorflow/translate/tf_mlir_translate_registration.cc

          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);
      if (!module_or.status().ok()) return nullptr;
      return std::move(module_or).value();
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 22:19:26 UTC 2024
    - 7.8K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tensorflow/utils/convert_tensor.cc

    absl::StatusOr<ElementsAttr> ConvertTensor(const Tensor& input_tensor,
                                               Builder* builder) {
      const auto& input_dtype = input_tensor.dtype();
      const auto& input_shape = input_tensor.shape();
      Type elt_type;
      TF_RETURN_IF_ERROR(ConvertDataType(input_dtype, *builder, &elt_type));
      SmallVector<int64_t, 4> shape;
      ConvertToMlirShape(input_shape, &shape);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Apr 26 09:37:10 UTC 2024
    - 20.5K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/lite/tf_to_tfl_flatbuffer.h

        bool use_splatted_constant, const std::vector<std::string>& extra_tf_opdefs,
        const GraphImportConfig& specs, absl::string_view debug_info_file,
        absl::string_view input_arrays, absl::string_view input_dtypes,
        absl::string_view input_shapes, absl::string_view output_arrays,
        absl::string_view control_output_arrays, llvm::SourceMgr* source_mgr,
        mlir::MLIRContext* context);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Apr 24 08:30:24 UTC 2024
    - 4.7K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/lite/tf_to_tfl_flatbuffer.cc

        return GraphdefToSplattedMlirTranslateFunction(
            file->getBuffer(), input_arrays, input_dtypes, input_shapes,
            output_arrays, control_output_arrays, graphdef_conversion_options,
            context);
      }
      return GraphdefToMlirTranslateFunction(file->getBuffer(), input_arrays,
                                             input_dtypes, input_shapes,
                                             output_arrays, control_output_arrays,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 03 18:01:23 UTC 2024
    - 23.8K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/tensorflow/transforms/region_control_flow_to_functional.cc

      // Build the function type. Region arguments and extern values together
      // become the function arguments, with region arguments going first.
      auto input_types = llvm::to_vector<4>(entry.getArgumentTypes());
      for (auto input : extern_values) input_types.push_back(input.getType());
    
      // Terminator operands and pass through extern values (if enabled) together
      // become the function return values.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 28.7K bytes
    - Viewed (0)
Back to top