- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 15 for int64Ptr (0.31 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/resource_op_lifting.cc
// body and condition. The sets of keys of the two maps must be the same. llvm::SmallDenseMap<int64_t, ResourceArgUseInfo> MergeArgResourceUseInfo( const llvm::SmallDenseMap<int64_t, ResourceArgUseInfo>& infos0, const llvm::SmallDenseMap<int64_t, ResourceArgUseInfo>& infos1) { llvm::SmallDenseMap<int64_t, ResourceArgUseInfo> result; for (const auto& entry : infos0) { auto info1_it = infos1.find(entry.getFirst());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 55.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/lower_tf.cc
SmallVector<int64_t> start_indices(input_rank, 0); SmallVector<int64_t> slice_sizes = reshaped_permuted_shape; SmallVector<int64_t> strides(input_rank, 1); auto crop_values = llvm::to_vector<4>(crops.getValues<APInt>()); for (int i = 0; i < block_rank; ++i) { int64_t crop_start = crop_values[i * 2].getSExtValue(); int64_t crop_end = crop_values[i * 2 + 1].getSExtValue();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 74.9K bytes - Viewed (0) -
tensorflow/c/c_api_test.cc
} } void TestEncodeDecode(int line, const std::vector<string>& data) { const int64_t n = data.size(); Status status; for (const std::vector<int64_t>& dims : std::vector<std::vector<int64_t>>{{n}, {1, n}, {n, 1}, {n / 2, 2}}) { // Create C++ Tensor Tensor src(tensorflow::DT_STRING, TensorShape(dims)); for (int64_t i = 0; i < src.NumElements(); ++i) { src.flat<tstring>()(i) = data[i]; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 15 03:35:10 UTC 2024 - 96.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/prepare_tf.cc
input_size - begin_shape[0] + 1 + absl::popcount(new_axis_mask); int64_t begin_mask = strided_slice_op.getBeginMask(); int64_t end_mask = strided_slice_op.getEndMask(); int64_t revised_begin_mask = 0; int64_t revised_end_mask = 0; int64_t revised_shrink_axis_mask = 0; int64_t revised_new_axis_mask = 0; SmallVector<int32_t, 4> padded_begin;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:49:50 UTC 2024 - 64.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/compose_uniform_quantized_type_pass.cc
// not contracted nor batched. int64_t GetFilterQuantizationDimension( const stablehlo::DotDimensionNumbersAttr& dot_dimension_numbers, const int64_t filter_rank) const { // Register all dimensions as candidates. auto seq_range = llvm::seq(int64_t{0}, filter_rank); SmallVector<int64_t> quantization_dimension_candidates(seq_range.begin(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 64.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/lower_static_tensor_list.cc
result_rank = element_type.getRank() + 1; int64_t leading_dim_v = -1; ElementsAttr element_attr; if (matchPattern(leading_dim, m_Constant(&element_attr))) { leading_dim_v = element_attr.getValues<APInt>()[0].getSExtValue(); } SmallVector<int64_t, 4> result_shape = {leading_dim_v}; ArrayRef<int64_t> shape = element_type.getShape();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 70.7K bytes - Viewed (0) -
tensorflow/compiler/jit/deadness_analysis.cc
protected: explicit Predicate(int64_t id) : id_(id) {} private: const int64_t id_; Predicate(const Predicate&) = delete; void operator=(const Predicate&) = delete; }; // Represents a logical conjunction of a set of predicates. class AndPredicate : public Predicate { public: explicit AndPredicate(int64_t id, std::vector<Predicate*> operands)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 60.4K bytes - Viewed (0) -
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) -
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) -
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)