Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 20 of 107 for pushBack (0.16 sec)

  1. tensorflow/compiler/mlir/quantization/stablehlo/passes/insert_calibration_statistics_saver.cc

        if (aggregator_ops_to_ignore.count(op.getId().str())) continue;
    
        ids.push_back(op.getId());
        calibration_methods.push_back(op.getCalibrationMethod());
        statistics_outputs.push_back(op.getMin());
        statistics_outputs.push_back(op.getMax());
        statistics_outputs.push_back(op.getHistogram());
      }
    }
    
    // Inserts a `CalibrationStatisticsSaverOp` to the end of the region.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 14 06:31:57 UTC 2024
    - 7.2K bytes
    - Viewed (0)
  2. tensorflow/c/experimental/ops/gen/common/case_format.cc

          if (wordStart) {
            result.push_back(delimiter);
          }
          wordStart = true;
          continue;
        }
        if (!from_snake && isupper(c)) {
          wordStart = true;
        }
    
        // add delimiter
        if (wordStart && toSnake && !inputStart) {
          result.push_back(delimiter);
        }
    
        // add the next letter from the input string (choosing upper/lower case)
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 21 09:51:28 UTC 2024
    - 3K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_config.cc

            const std::vector<std::string> indices =
                absl::StrSplit(node_specification, '-');
            for (const std::string& cur_index : indices) {
              custom_op_map[node_name].quantizable_input_indices.push_back(
                  std::stoi(cur_index));
            }
            break;
          }
          case CustomOpUpdateOptions::kWeightOnly:
            custom_op_map[node_name].is_weight_only =
                GetBooleanSpecs(node_specification);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 05 07:39:40 UTC 2024
    - 5.9K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/lite/experimental/tac/execution_metadata_exporter.cc

      if (!main_fn) return std::string("");
    
      flatbuffers::FlatBufferBuilder fb_builder;
      std::vector<mlir::func::FuncOp> funcs;
      funcs.push_back(main_fn);
      module.walk([&](mlir::func::FuncOp fn) {
        if (fn != main_fn) {
          funcs.push_back(fn);
        }
      });
    
      // Populate the hardware metadata.
      // And collect the hardwares used.
      std::map<std::string, uint8_t> hardware_map;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 06:11:34 UTC 2024
    - 7.5K bytes
    - Viewed (0)
  5. tensorflow/cc/client/client_session.cc

      output_tensor_names.reserve(fetch_outputs.size());
      for (auto const& output : fetch_outputs) {
        output_tensor_names.push_back(output.name());
      }
      std::vector<string> target_node_names;
      target_node_names.reserve(run_outputs.size());
      for (auto const& output : run_outputs) {
        target_node_names.push_back(output.node()->name());
      }
      TF_RETURN_IF_ERROR(impl()->MaybeExtendGraph());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 28 09:04:10 UTC 2024
    - 7.1K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tensorflow/transforms/executor_tpuv1_outline_tpu_island.cc

        func_operand_types.reserve(operands.size());
        for (Value operand : operands)
          func_operand_types.push_back(operand.getType());
    
        // Function results are the yield operands
        SmallVector<Type, 16> func_result_types;
        for (Value operand : island_op.GetYield().getOperands())
          func_result_types.push_back(operand.getType());
        FunctionType func_type =
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 8.7K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tensorflow/transforms/xla_rewrite.cc

      for (const Value &arg : cluster_func_op.getOperands()) {
        if (!mlir::isa<TF::ResourceType>(getElementTypeOrSelf(arg.getType()))) {
          non_resource_args.push_back(arg);
          if (has_resources) in_order = false;
        } else {
          resource_args.push_back(arg);
          has_resources = true;
        }
      }
    
      if (!in_order) {
        // Functions do not get reused in practice, so skip the check for if the
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 4.2K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/tensorflow/transforms/set_tpu_infeed_layout.cc

        for (const mlir::Type &t : types) {
          if (mlir::isa<mhlo::TokenType>(t)) continue;
          auto layout = GetTPUInfeedLayout({t}, rewriter);
          if (failed(layout)) return failure();
          v.push_back(layout.value());
        }
        ArrayRef<Attribute> shape(v);
        return rewriter.getArrayAttr(shape);
      } else if (mlir::isa<TupleType>(types[0])) {
        auto tuple_type = mlir::dyn_cast<TupleType>(types[0]);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 6.1K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/quantization/tensorflow/cc/constant_fold.cc

        inputs.push_back(preceding_const_op.getValue());
      }
    
      SmallVector<Attribute> result_values;
      if (failed(TF::EvaluateOperation(op, inputs, result_values))) {
        return failure();
      }
    
      results.clear();
      builder.setInsertionPointAfter(op);
      for (const auto& result_value : result_values) {
        results.push_back(builder.create<TF::ConstOp>(op->getLoc(), result_value));
      }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 5K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/lite/tests/flatbuffer2mlir/importer_test_min_max.cc

            input_tensor->quantization->scale.clear();
            input_tensor->quantization->zero_point.clear();
            input_tensor->quantization->min.push_back(-1.0);
            input_tensor->quantization->max.push_back(1.0);
    
            auto& output_tensor = sub_graph->tensors[op->outputs[0]];
            auto shape = output_tensor->shape;
            output_tensor->quantization->scale.clear();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 21 18:21:50 UTC 2024
    - 6.8K bytes
    - Viewed (0)
Back to top