Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 20 of 20 for int64Ptr (1.04 sec)

  1. tensorflow/compiler/mlir/quantization/tensorflow/passes/quantize_composite_functions.cc

      // so not passing it during conversion. This design might change if needed.
      ArrayRef<double> scales = qtype.getScales();
      ArrayRef<int64_t> zero_points = qtype.getZeroPoints();
      const int num_channels = scales.size();
      TensorType scale_type = RankedTensorType::get(
          {static_cast<int64_t>(num_channels)}, rewriter.getF32Type());
      TensorType zero_point_type = scale_type.clone(rewriter.getI32Type());
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 54.5K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/mark_for_compilation_pass.cc

                    .session_metadata()
                    .name()
              : ""}
          .Run();
    }
    
    std::atomic<int64_t>* GetPointerToFuel(int64_t initial_value) {
      static std::atomic<int64_t>* fuel = [&]() {
        std::atomic<int64_t>* fuel = new std::atomic<int64_t>;
        *fuel = initial_value;
        return fuel;
      }();
    
      return fuel;
    }
    
    }  // anonymous namespace
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 21 12:19:41 UTC 2024
    - 85.3K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/transforms/sparsecore/embedding_pipelining.cc

      a.insert(a.end(), b.begin(), b.end());
    }
    
    template <typename Vector>
    void Append(Vector& a, const Vector& b) {
      a.insert(a.end(), b.begin(), b.end());
    }
    
    int64_t GetNumOps(func::FuncOp func) {
      int64_t num_ops = 0;
      for (auto it = func.begin(); it != func.end(); ++it) ++num_ops;
      return num_ops;
    }
    
    std::vector<Value> ResultsAsVector(Operation* op) {
      std::vector<Value> vec;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 92.9K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/lite/quantization/lite/quantize_model_test.cc

      const std::vector<float>& bias_scales = bias_tensor->quantization->scale;
      const std::vector<float>& weights_scales =
          weights_tensor->quantization->scale;
      const std::vector<int64_t>& weights_zero_points =
          weights_tensor->quantization->zero_point;
      const int out_channel_size = weights_tensor->shape[0];
      ASSERT_THAT(bias_scales, SizeIs(out_channel_size));
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 23:15:24 UTC 2024
    - 73.9K bytes
    - Viewed (0)
  5. tensorflow/c/experimental/filesystem/filesystem_interface.h

    } TF_TransactionToken;
    
    // The named union is needed here (as opposed to
    // inside the `TF_Filesystem_Option_Value` struct)
    // as MSVC does not recognize `typeof`.
    typedef union TF_Filesystem_Option_Value_Union {
      int64_t int_val;
      double real_val;
      struct {
        char* buf;
        int buf_length;
      } buffer_val;
    } TF_Filesystem_Option_Value_Union;
    
    typedef struct TF_Filesystem_Option_Value {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 27 17:36:54 UTC 2022
    - 53.1K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tf2xla/internal/passes/extract_outside_compilation.cc

                  "ranked tensors.";
      ArrayRef<int64_t> in_shape = ranked_type.getShape();
      if (in_shape.empty() || in_shape[0] < 0) {
        return context_op->emitOpError()
               << "A map_outside_compilation op's input and output shapes must "
                  "have rank at least one and the first dimension must be known.";
      }
      int64_t split_size = in_shape[0] / num_cores_per_replica;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 30 21:25:12 UTC 2024
    - 68.3K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/lite/flatbuffer_import.cc

          {min_maxs[0], min_maxs[1]});
      mlir::ElementsAttr axis_stats;
      mlir::IntegerAttr axis;
      if (mins.size() > 1) {
        llvm::SmallVector<int64_t, 4> axis_stats_shape{
            static_cast<int64_t>(mins.size()), 2};
        axis_stats = mlir::DenseFPElementsAttr::get(
            tensorflow::GetTypeFromTFTensorShape(axis_stats_shape, b.getF32Type()),
            min_maxs);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 21 18:21:50 UTC 2024
    - 66.8K bytes
    - Viewed (0)
  8. tensorflow/c/eager/c_api_test.cc

      TFE_ContextOptions* opts = TFE_NewContextOptions();
      TFE_Context* ctx = TFE_NewContext(opts, status);
      ASSERT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
      TFE_DeleteContextOptions(opts);
    
      std::vector<int64_t> dims(4, 1);
      TFE_Op* op = TFE_NewOp(ctx, "AvgPool", status);
      ASSERT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
    
      TF_Tensor* tensor =
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Aug 03 20:50:20 UTC 2023
    - 94.6K bytes
    - Viewed (0)
  9. tensorflow/compiler/jit/deadness_analysis_test.cc

      EXPECT_EQ(predicate_map[ControlOutputFor(id_true)], "*cond_ref:0");
    }
    
    void CreateSwitchN(const Scope& scope, Input data, Input output_index,
                       int64_t num_outs, OutputList* outputs) {
      if (!scope.ok()) return;
      auto _data = ops::AsNodeOut(scope, data);
      if (!scope.ok()) return;
      auto _output_index = ops::AsNodeOut(scope, output_index);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 06:59:07 UTC 2024
    - 51.6K bytes
    - Viewed (0)
  10. tensorflow/compiler/jit/mark_for_compilation_pass_test.cc

      Output delta = ops::Placeholder(scope.WithOpName("delta"), DT_FLOAT);
    
      ops::AssignAddVariableOp increment_op(
          scope.WithOpName("IncrementIteration"), iter,
          ops::Const(scope.WithOpName("one"), static_cast<int64_t>(1)));
    
      ops::AssignAddVariableOp weights_0_update_op(
          scope.WithOpName("weights_0_update"), weights_0, delta);
      ops::AssignAddVariableOp weights_1_update_op(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 14 10:11:10 UTC 2024
    - 79.6K bytes
    - Viewed (0)
Back to top