Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 103 for Auto (0.07 sec)

  1. tensorflow/compiler/mlir/lite/flatbuffer_export.cc

        attr = cst.getValue();
      } else if (auto cst = dyn_cast<tfl::QConstOp>(inst)) {
        attr = cst.getValue();
      } else if (auto cst = dyn_cast<mlir::stablehlo::ConstantOp>(inst)) {
        attr = cst.getValue();
      } else if (auto cst = dyn_cast<mlir::vhlo::ConstantOpV1>(inst)) {
        mlir::VhloToStablehloTypeConverter vhlo_type_converter;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 21:41:49 UTC 2024
    - 164.5K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/quantization/lite/quantize_model_test.cc

                                        const OperatorT& quant_op, int idx) {
      const auto& builtin_code =
          GetBuiltinCode(quant_model.operator_codes[quant_op.opcode_index].get());
      for (const auto& expected_op : expected_graph.operators) {
        const auto& op_code =
            expected_model.operator_codes[expected_op->opcode_index].get();
        const auto& expected_code = GetBuiltinCode(op_code);
        if (expected_code == builtin_code) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 23:15:24 UTC 2024
    - 73.9K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tf2xla/api/v2/tf_executor_to_graph.cc

    }
    
    void Exporter::UseOriginalFunctionNames(NodeDef& node_def) {
      if (!configs_.export_original_tf_func_name) return;
    
      auto& attrs = *node_def.mutable_attr();
    
      auto try_use_original_func_name = [this](std::string* name) {
        if (auto func = symbol_table_.lookup<FuncOp>(*name)) {
          if (auto original_func_name =
                  func->getAttrOfType<mlir::StringAttr>("tf._original_func_name")) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 06 23:04:51 UTC 2024
    - 35.2K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/lite/quantization/lite/quantize_weights_test.cc

           subgraph_idx++) {
        const auto quantized_graph = output_model->subgraphs()->Get(subgraph_idx);
        const auto float_graph = model_->subgraphs()->Get(subgraph_idx);
        ASSERT_EQ(quantized_graph->tensors()->size(),
                  float_graph->tensors()->size());
        std::vector<int> used_tensors;
        for (size_t i = 0; i < quantized_graph->tensors()->size(); i++) {
          const auto quant_tensor = quantized_graph->tensors()->Get(i);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 23:15:24 UTC 2024
    - 32.3K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/transforms/lower_static_tensor_list.cc

        auto guard = OpBuilder::InsertionGuard(*rewriter);
        auto inputs = branch_func.getFunctionType().getInputs();
        Block *block = rewriter->createBlock(
            &branch_func.getBody(), branch_func.begin(), inputs,
            SmallVector<Location>(inputs.size(), branch_func.getLoc()));
    
        auto input_shape = block->getArgument(1);
        auto size_diff = block->getArgument(2);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 20:00:43 UTC 2024
    - 70.7K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/lite/experimental/tac/execution_metadata_exporter_test.cc

      mlir::MLIRContext context(registry);
      auto module = mlir::OwningOpRef<mlir::ModuleOp>(
          mlir::parseSourceString<mlir::ModuleOp>(kMLIR, &context));
      auto module_op = module.get();
      auto serialized_result_fb = ExportRuntimeMetadata(module_op);
      const auto* result = GetRuntimeMetadata(serialized_result_fb.value().c_str());
      const auto* expected = GetRuntimeMetadata(kExpectedFB.c_str());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 06:11:34 UTC 2024
    - 6K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc

    // resolved
    LogicalResult PrecheckForXlaConvV2Op(XlaConvV2Op op) {
      auto input_tensor = op.getLhs();
      auto kernel_tensor = op.getRhs();
      auto window_strides = op.getWindowStrides();
      auto padding = op.getPadding();
      auto lhs_dilation = op.getLhsDilation();
      auto rhs_dilation = op.getRhsDilation();
      auto feature_group_count = op.getFeatureGroupCount();
      int64_t batch_group_count = op.getBatchGroupCount();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Jun 08 07:28:49 UTC 2024
    - 134.1K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/lite/sparsity/sparsify_model_test.cc

      auto output_fbm = tflite::FlatBufferModel::BuildFromBuffer(
          reinterpret_cast<const char*>(output_builder.GetCurrentBufferPointer()),
          output_builder.GetSize());
      tflite::ModelT output_model;
      output_fbm->GetModel()->UnPackTo(&output_model);
    
      // Extract output metadata
      std::map<std::string, std::string> output_metadata;
      for (const auto& metadata : output_model.metadata) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Jun 10 20:16:40 UTC 2024
    - 2.9K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc

        Location loc = op.getLoc();
        auto params = op.getParams();
        auto params_ty = mlir::dyn_cast<RankedTensorType>(params.getType());
        auto indices = op.getIndices();
        auto indices_ty = mlir::dyn_cast<RankedTensorType>(indices.getType());
        auto params_rank = params_ty.getRank();
        auto indices_rank = indices_ty.getRank();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 20:00:43 UTC 2024
    - 291.8K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/tensorflow/transforms/decompose_optionals.cc

        //   ...
        // }
        auto ret = f.getBody().front().getTerminator();
        std::vector<Type> argument_types;
        for (auto arg : f.getBody().front().getArguments()) {
          argument_types.push_back(arg.getType());
        }
        std::vector<Type> return_types;
        for (auto ret_op : ret->getOperands()) {
          return_types.push_back(ret_op.getType());
        }
        auto newType =
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 21:18:05 UTC 2024
    - 9.8K bytes
    - Viewed (0)
Back to top