- Sort Score
- Result 10 results
- Languages All
Results 81 - 90 of 453 for int64Opt (0.15 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/constant_fold.cc
} return size; }; int64_t results_size = get_size(inst->getResultTypes()); int64_t operands_size = get_size(inst->getOperandTypes()); constexpr int kSizeFactor = 2; // TODO(b/233827625): Remove TF_DISABLE_CONSTANT_FOLDING macro. #ifdef TF_DISABLE_CONSTANT_FOLDING constexpr int64_t kResultsSizeThreshold = 0; #else constexpr int64_t kResultsSizeThreshold = (1 << 23); // 1 MB #endif
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.3K bytes - Viewed (0) -
tensorflow/cc/experimental/base/public/tensor.h
// a TFRT backed tensor. // TODO(bmzhao): Add benchmarks on overhead for this function; we can // consider using int64_t* + length rather than vector. static Tensor FromBuffer(TF_DataType dtype, const std::vector<int64_t>& shape, void* data, size_t len, DeleterCallback deleter, Status* status);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 28 20:10:33 UTC 2020 - 6.3K bytes - Viewed (0) -
tensorflow/c/ops.cc
} void TF_ShapeInferenceContextDim(TF_ShapeInferenceContext* ctx, TF_ShapeHandle* shape_handle, int64_t i, TF_DimensionHandle* result) { int64_t rank = TF_ShapeInferenceContextRank(ctx, shape_handle); auto* cc_result = reinterpret_cast<DimensionHandle*>(result); if (i < -rank || i >= rank) { *cc_result = DimensionHandle(); return; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jul 28 22:41:35 UTC 2022 - 10.9K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_device_ops.h
.HostMemory("output") \ .TypeConstraint<int64_t>("out_type") \ .TypeConstraint("T", TYPES), \ ShapeOp<int64_t>); \ REGISTER_KERNEL_BUILDER(Name("ShapeN") \
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Nov 23 19:28:25 UTC 2021 - 17.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/cc/permutation_test.cc
TEST(PermutationTest, PermuteEmptyArray) { const SmallVector<int> permutation_result = Permute<int>(SmallVector<int>{}, SmallVector<int64_t>{}); EXPECT_THAT(permutation_result, IsEmpty()); } TEST(PermutationTest, PermuteOneElement) { const SmallVector<int> single_element_array = {8}; const SmallVector<int64_t> permutation = {0}; const SmallVector<int> permutation_result = Permute<int>(single_element_array, permutation);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 22 00:14:00 UTC 2024 - 2.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/set_tpu_infeed_layout.cc
std::iota(minor_to_major.begin(), minor_to_major.end(), 0); std::sort(minor_to_major.begin(), minor_to_major.end(), [=](int64_t a, int64_t b) { int64_t da = t.getDimSize(a); int64_t db = t.getDimSize(b); return da > db || (da == db && a > b); }); } std::vector<Attribute> elements;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 6.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc
std::optional<RankedTensorType> InferXlaConvOutputShape( llvm::SmallVector<int64_t> input_tensor_dims, llvm::SmallVector<int64_t> kernel_tensor_dims, llvm::SmallVector<int64_t> window_strides, llvm::SmallVector<std::pair<int64_t, int64_t>> paddings, llvm::SmallVector<int64_t> lhs_dilations, llvm::SmallVector<int64_t> rhs_dilations, int64_t batch_group_count, xla::ConvolutionDimensionNumbers dnums, Type element_type) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jun 08 07:28:49 UTC 2024 - 134.1K bytes - Viewed (0) -
tensorflow/c/eager/c_api_experimental_reader.h
TFE_MonitoringNewCounterReader(const char* name); // Reads the value of a counter that was created with 0 labels. TF_CAPI_EXPORT extern int64_t TFE_MonitoringReadCounter0( TFE_MonitoringCounterReader*); // Reads the value of specific cell of a counter that was created with 1 label. TF_CAPI_EXPORT extern int64_t TFE_MonitoringReadCounter1( TFE_MonitoringCounterReader*, const char* label_value); #ifdef __cplusplus } /* end extern "C" */
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 20 03:14:47 UTC 2023 - 2.3K bytes - Viewed (0) -
tensorflow/c/tf_tensor.h
TF_CAPI_EXPORT extern void TF_SetShape(TF_Tensor* tensor, const int64_t* dims, int num_dims); // Return the number of dimensions that the tensor has. TF_CAPI_EXPORT extern int TF_NumDims(const TF_Tensor*); // Return the length of the tensor in the "dim_index" dimension. // REQUIRES: 0 <= dim_index < TF_NumDims(tensor) TF_CAPI_EXPORT extern int64_t TF_Dim(const TF_Tensor* tensor, int dim_index);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 06 16:40:30 UTC 2024 - 6.3K bytes - Viewed (0) -
tensorflow/c/experimental/saved_model/core/test_utils.cc
std::vector<DataType> result; result.reserve(set.size()); for (DataType dt : set) { result.push_back(dt); } return result; } std::vector<std::vector<int64_t>> InterestingShapes() { std::vector<std::vector<int64_t>> interesting_shapes; interesting_shapes.push_back({}); // Scalar interesting_shapes.push_back({10}); // 1D Vector interesting_shapes.push_back({3, 3}); // 2D Matrix
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Sep 08 20:13:32 UTC 2023 - 5.8K bytes - Viewed (0)