- Sort Score
- Result 10 results
- Languages All
Results 71 - 80 of 453 for int64Opt (0.26 sec)
-
tensorflow/compiler/mlir/quantization/stablehlo/passes/nchw_convolution_to_nhwc.cc
/*inputSpatialDimensions=*/SmallVector<int64_t>{1, 2}, /*kernelInputFeatureDimension=*/2, /*kernelOutputFeatureDimension=*/3, /*kernelSpatialDimensions=*/SmallVector<int64_t>{0, 1}, /*outputBatchDimension=*/0, /*outputFeatureDimension=*/3, /*outputSpatialDimensions=*/SmallVector<int64_t>{1, 2}); // Determine the shape of the output tensor: [b, f, 0, 1] => [b, 0, 1, f]
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/transforms/cost_model.cc
constexpr float kRequantCost = 2.0; // TODO(renjieliu): Ideally this should consider different kinds of SOCs as // well. // Get total bytes transferred. int64_t GetTransferredTensorBytes(func::CallOp from_graph, func::CallOp to_graph) { int64_t total_size_transferred = 0; for (auto input : to_graph.getOperands()) { Operation* input_op = input.getDefiningOp();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.3K bytes - Viewed (0) -
tensorflow/c/eager/c_api_test_util.cc
return th; } TFE_TensorHandle* DoubleTestMatrixTensorHandle(TFE_Context* ctx) { int64_t dims[] = {2, 2}; double data[] = {1.0, 2.0, 3.0, 4.0}; TF_Status* status = TF_NewStatus(); TF_Tensor* t = TFE_AllocateHostTensor(ctx, TF_DOUBLE, &dims[0], sizeof(dims) / sizeof(int64_t), status); memcpy(TF_TensorData(t), &data[0], TF_TensorByteSize(t));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 22:37:46 UTC 2024 - 23.5K bytes - Viewed (0) -
tensorflow/c/experimental/gradients/tape/tape_operation.cc
} Status TapeOperation::SetAttrIntList(const char* attr_name, const int64_t* values, int num_values) { forward_op_.attrs.Set( attr_name, gtl::ArraySlice<const int64_t>( reinterpret_cast<const int64_t*>(values), num_values)); return parent_op_->SetAttrIntList(attr_name, values, num_values); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 28 06:16:45 UTC 2024 - 9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_utils.cc
if (num_bits >= 8) { return type; } int64_t qmin = QType::getDefaultMinimumForInteger(is_signed, num_bits); int64_t qmax = QType::getDefaultMaximumForInteger(is_signed, num_bits); if (narrow_range) { qmin += 1; } const int64_t storage_type_min = type.getStorageTypeMin(); const int64_t storage_type_max = type.getStorageTypeMax(); const double rate =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 08 02:10:16 UTC 2024 - 43.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/utils.cc
RankedTensorType::get(static_cast<int64_t>(attr.size()), IntegerType::get(attr.getContext(), 64)); return DenseIntElementsAttr::get(ty, attr.getValue()); } DenseIntElementsAttr GetI64ElementsAttr(ArrayRef<int64_t> values, Builder* builder) { RankedTensorType ty = RankedTensorType::get( {static_cast<int64_t>(values.size())}, builder->getIntegerType(64));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 1.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/convert_tf_quant_ops_to_mhlo.cc
const int64_t output_size = (lhs_size_dilated + stride - 1) / stride; const int64_t total_padding = std::max( (output_size - 1) * stride + rhs_size_dilated - lhs_size_dilated, static_cast<int64_t>(0)); const int64_t padding_begin = total_padding / 2; const int64_t padding_end = total_padding - padding_begin;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 30.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/stack_ops_decomposition.cc
llvm::function_ref<std::optional<Type>(int64_t)> arg_to_stack_type, llvm::function_ref<void(ArrayRef<BlockArgument>)> handle_new_size_vars = nullptr) { auto new_input_types = llvm::to_vector<8>(func.getFunctionType().getInputs()); auto size_var_type = GetSizeVarType(OpBuilder(func)); int64_t original_arg_count = new_input_types.size(); for (int64_t i = 0; i < original_arg_count; ++i) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 23.8K bytes - Viewed (0) -
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/compiler/mlir/quantization/common/attrs_and_constraints.h
// permutation of `kNchwToNhwcPermutation`. inline constexpr std::array<int64_t, 4> kNhwcToNchwPermutation = {0, 3, 1, 2}; // Permutation from the NCHW tensor format to NHWC. This is an inverse // permutation of `kNchwToNhwcPermutation`. inline constexpr std::array<int64_t, 4> kNchwToNhwcPermutation = {0, 2, 3, 1}; // Permutation from the OIHW (== (output features, input features, height,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 9.9K bytes - Viewed (0)