- Sort Score
- Result 10 results
- Languages All
Results 21 - 30 of 345 for SmallVector (0.31 sec)
-
tensorflow/compiler/mlir/lite/utils/lstm_utils.h
Value output_layer_norm_coefficients_; mlir::TFL::LSTMOp lstm_; Value none_; SmallVector<int64_t, 1> bias_slice_shape_; SmallVector<int64_t, 1> bias_size_values_; SmallVector<int64_t, 2> weight_slice_shape_; SmallVector<int64_t, 2> weight_slice_size_input_values_; SmallVector<int64_t, 2> weight_slice_size_recurrent_values_; OpBuilder builder_; };
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jun 03 00:14:05 UTC 2023 - 7.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/shape_inference_pass.cc
: input_shapes_(input_shapes) {} void runOnOperation() override { // Parse `input_arg_shapes_` if provided (test only) SmallVector<ArrayRef<int64_t>> input_shapes_vec; absl::StatusOr<SmallVector<SmallVector<int64_t>>> parsed_shapes; if (!input_arg_shapes_.empty()) { parsed_shapes = ParseArgumentShapes(input_arg_shapes_); if (!parsed_shapes.ok()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 12:49:45 UTC 2024 - 3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo_conversions/dot_general.cc
// TFL::UnsortedSegmentProdOp. llvm::SmallVector<int32_t, 4> flattened_out_segids = llvm::SmallVector<int32_t, 4>(operand_rank, static_cast<int32_t>(-1)); for (int64_t i : dot_dimensions_info.out_dimensions().AxesArray()) { flattened_out_segids[i] = 0; } llvm::SmallVector<int32_t, 4> flattened_contracting_segids = llvm::SmallVector<int32_t, 4>(operand_rank, static_cast<int32_t>(-1));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 19.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/utils/tf_to_xla_attribute_utils.cc
StringAttr conv_padding, Value &padding, int num_dims) { Value zero_rank1 = CreateConstValue<int32_t>(builder, loc, {1}, {0}); SmallVector<Value> temp_padding_values{zero_rank1, zero_rank1}; auto reshape_op = [&](Value value, const SmallVector<int64_t> &shape) { const int64_t rank = shape.size(); return builder.create<TF::ReshapeOp>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 13.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/xla_sharding_util.h
mlir::OpBuilder* builder, llvm::SmallVectorImpl<llvm::SmallVector<mlir::Value, 4>>* input_list); // Extracts a list of OpSharding that represent output sharding configuration of // `tf_device.cluster`. mlir::LogicalResult ParseAndValidateOutputSharding( int num_cores_per_replica, mlir::tf_device::ClusterFuncOp cluster_func, mlir::SmallVector<xla::OpSharding, 4>* output_sharding_list);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 28 22:18:34 UTC 2024 - 6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_saved_model_freeze_variables.cc
// in 'arguments_to_erase'. template <typename T, typename U> T GetUpdatedWhileOp(T while_op, const U& argument_types, const llvm::SmallVector<unsigned, 4>& arguments_to_erase) { OpBuilder builder(while_op); llvm::SmallVector<Type, 4> new_operand_types; llvm::SmallVector<Value> new_operands; auto operands = while_op->getOperands(); const int num_operands = while_op->getNumOperands();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 09:56:53 UTC 2024 - 19.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/cc/permutation.h
#include <type_traits> #include "llvm/ADT/ArrayRef.h" // IWYU pragma: keep; required to include the definition of ArrayRef #include "llvm/ADT/STLExtras.h" #include "llvm/ADT/SmallVector.h" // IWYU pragma: keep; required to include the definition of SmallVector #include "mlir/Support/LLVM.h" // from @llvm-project namespace mlir::quant { // Permutes `values` with `permutation`. Returns the permuted values. Sizes of
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 22 00:14:00 UTC 2024 - 1.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/fold_constant_transpose.cc
SmallVector<float> TransposeValues(const ArrayRef<float> values) const { SmallVector<float> transposed_values(values.size()); SmallVector<int64_t> current_indices = {}; TransposeRecursively(values, transposed_values, current_indices); return transposed_values; } // Returns the shape after permutation. SmallVector<int64_t> GetTargetShape() const { return target_shape_; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo_conversions/util.cc
} else { // Convert I64 const array to I32. llvm::SmallVector<int32_t> const_i32_vec; for (auto element : const_value) { const_i32_vec.push_back(static_cast<int32_t>(element)); } const_value_raw = rewriter.getI32TensorAttr(const_i32_vec); } Value result_const = builder.create<TF::ConstOp>(const_value_raw); return result_const; } llvm::SmallVector<int64_t> GetInversePermutationArray(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_rewrite_pass.cc
void AssignDevicesToReplicate( tf_device::ReplicateOp replicate, llvm::ArrayRef<llvm::SmallVector<tensorflow::TPUDeviceAndHost, 8>> tpu_devices, OpBuilder* builder) { if (!replicate) return; const int num_replicas = tpu_devices.size(); const int num_cores_per_replica = tpu_devices.front().size(); llvm::SmallVector<NamedAttribute, 8> device_attrs; for (int core = 0; core < num_cores_per_replica; ++core) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 21:25:12 UTC 2024 - 29.7K bytes - Viewed (0)