- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 12 for int64Ptr (0.24 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc
auto can_convert_to_bilinear = [](bool align_corners, int64_t dilation, int64_t padding, int64_t stride, int64_t input_spatial, int64_t output_spatial) { int64_t input_spatial_size = align_corners ? input_spatial - 1 : input_spatial; int64_t output_spatial_size =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 154.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/uniform_quantized_stablehlo_to_tfl_pass.cc
// (spatial_filter_shape[i]-1) * dilation_rate[i]) / strides[i]) auto get_output_dim_for_valid_padding = [](int64_t input_dim, int64_t dilation_dim, int64_t kernel_dim, int64_t stride_dim) -> int64_t { return std::ceil((input_dim - (kernel_dim - 1) * dilation_dim) / stride_dim); }; return output_height ==
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 22 09:00:19 UTC 2024 - 99.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/ir/tfl_ops.cc
int64_t x_row_dim = x_shape[x_shape.size() - 2]; int64_t x_col_dim = x_shape[x_shape.size() - 1]; int64_t y_row_dim = y_shape[y_shape.size() - 2]; int64_t y_col_dim = y_shape[y_shape.size() - 1]; int64_t out_row_dim = output_shape[output_shape.size() - 2]; int64_t out_col_dim = output_shape[output_shape.size() - 1];
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 169.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc
MutableArrayRef<int64_t> input_shape, ArrayRef<int64_t> sparse_begin, ArrayRef<int64_t> sparse_end, ArrayRef<int64_t> sparse_strides, int32_t begin_mask, int32_t end_mask, int32_t ellipsis_mask, int32_t new_axis_mask, int32_t shrink_axis_mask, SmallVectorImpl<int64_t> *begin, SmallVectorImpl<int64_t> *end, SmallVectorImpl<int64_t> *stride) { int64_t num_sparse_indices = sparse_begin.size();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 22:07:10 UTC 2024 - 170.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_a_m.cc
int64_t x_row_dim = x_shape[x_shape.size() - 2]; int64_t x_col_dim = x_shape[x_shape.size() - 1]; int64_t y_row_dim = y_shape[y_shape.size() - 2]; int64_t y_col_dim = y_shape[y_shape.size() - 1]; int64_t out_row_dim = output_shape[output_shape.size() - 2]; int64_t out_col_dim = output_shape[output_shape.size() - 1]; int64_t expected_out_row_dim = op.getAdjX() ? x_col_dim : x_row_dim;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 146.7K 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/compiler/mlir/lite/flatbuffer_export.cc
std::unordered_map<uint64_t, std::pair<int64_t, int64_t>> hashcode_to_pos; // Buffer data should be exported only once. assert(!buffer_data_exported_); auto it = buffer_data_map_.begin(); while (it != buffer_data_map_.end()) { std::string buffer = it->second; int64_t index = it->first; int64_t offset = result.size(); int64_t size = buffer.size();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:41:49 UTC 2024 - 164.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/optimize.cc
static std::pair<int64_t, int64_t> GetBiasDimAndSliceSize( ArrayRef<int64_t> filter_shape, AffineOpType op) { // Channel dimension index is specified as op property auto channel_index_iter = filter_shape.begin(); std::advance(channel_index_iter, op.GetChannelDimIndex()); // The slide size is the size of the data in higher dimensions. int64_t slice_size =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 00:40:15 UTC 2024 - 102.3K bytes - Viewed (0) -
tensorflow/c/c_api.cc
int64_t value) { desc->node_builder.Attr(attr_name, static_cast<int64_t>(value)); } void TF_SetAttrIntList(TF_OperationDescription* desc, const char* attr_name, const int64_t* values, int num_values) { desc->node_builder.Attr( attr_name, ArraySlice<const int64_t>( reinterpret_cast<const int64_t*>(values), num_values)); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 15 03:35:10 UTC 2024 - 102.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/import_model.cc
} if (c->Rank(s0) != c->Rank(s1)) { return false; } for (int i = 0; i < c->Rank(s0); ++i) { if (!c->Dim(s0, i).SameHandle(c->Dim(s1, i))) { int64_t val0 = c->Value(c->Dim(s0, i)); int64_t val1 = c->Value(c->Dim(s1, i)); // Negative value is treated as unknown so all negative values indicate // the same dimension.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 183.2K bytes - Viewed (0)