- Sort Score
- Result 10 results
- Languages All
Results 41 - 50 of 345 for SmallVector (0.6 sec)
-
tensorflow/compiler/mlir/tfrt/ir/tfrt_fallback_common.cc
mlir::IntegerAttr op_key; mlir::IntegerAttr cost; mlir::StringAttr device; mlir::StringAttr op_name; mlir::SymbolRefAttr f; llvm::SmallVector<mlir::OpAsmParser::UnresolvedOperand, 4> in_chains; llvm::SmallVector<mlir::OpAsmParser::UnresolvedOperand, 4> operands; mlir::NamedAttrList op_attrs; mlir::NamedAttrList op_func_attrs; auto loc = parser.getNameLoc(); if (options.has_chain &&
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/tpu_rewrite_device_util_test.cc
parsed_devices->push_back(parsed_name); } return true; } using DeviceNames = llvm::SmallVector<std::string, 8>; struct ParameterizedDeviceSetTest : ::testing::TestWithParam<std::tuple<DeviceNames, std::string>> {}; TEST_P(ParameterizedDeviceSetTest, BadDeviceSet) { llvm::SmallVector<Device, 8> devices; ASSERT_TRUE(DeviceNamesToParsedNames(std::get<0>(GetParam()), &devices)); std::string topology_attr;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Apr 26 09:37:10 UTC 2024 - 46.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/dynamic_shape_utils.cc
#include "llvm/ADT/SmallVector.h" namespace tensorflow { llvm::SmallVector<int64_t> ConvertTFShapeToMlir( llvm::ArrayRef<int64_t> shapes) { return llvm::to_vector(llvm::map_range(shapes, [](int64_t shape) { return shape == kTFDynamicSize ? mlir::ShapedType::kDynamic : shape; })); } llvm::SmallVector<int64_t> ConvertMlirShapeToTF( llvm::ArrayRef<int64_t> shapes) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Nov 21 16:21:18 UTC 2022 - 1.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc
// Compute ConvDimensionNumbers, dilation, and padding. SmallVector<int64_t, num_spatial_dims> spatial_dims; SmallVector<int64_t, num_spatial_dims> kernel_spatial_dims; SmallVector<int64_t, num_spatial_dims> rhs_dilation; SmallVector<int64_t, num_spatial_dims * 2> paddings; SmallVector<int64_t, num_spatial_dims> window_strides;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 291.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/const_tensor_utils.cc
// bytes that represent host endianness values. // The read_size parameter is present to allow reading both float16 and float32 // without a case split. template <typename T> llvm::SmallVector<mlir::APInt> ReadAsHostEndian(ArrayRef<uint8_t> bytes) { llvm::SmallVector<mlir::APInt> ret; size_t read_size = sizeof(T); int bytes_len = bytes.size(); assert(bytes_len % read_size == 0); int elem_count = bytes_len / read_size;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 07 23:04:40 UTC 2024 - 16.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_space_to_depth_pass.cc
MLIRContext* context = backprop.getContext(); SmallVector<int64_t, 4> values = {1, 1, 1, 1}; auto attrs = llvm::map_range(values, [context](int64_t v) -> Attribute { return IntegerAttr::get(IntegerType::get(context, 64), APInt(64, v)); }); auto strides = ArrayAttr::get(context, llvm::to_vector<4>(attrs)); // new result type. SmallVector<int64_t, 4> new_shape(new_filter_shape.begin(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 29.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/quantization_patterns.h
private: // Collects all candidate ops for quantization, which are the // `dequantize_op`'s users. FailureOr<SmallVector<Operation*>> CollectCandidateOps( DequantizeOpT dequantize_op) const { auto users = dequantize_op->getResult(0).getUsers(); return SmallVector<Operation*>(users.begin(), users.end()); } // Collects all candidate ops for quantization, which is the operand of
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/lift_as_function_call.cc
SmallVector<NamedAttribute> attributes; return LiftAsFunctionCall(builder, location, call_op_type, func_name, arguments, results, attributes); } SmallVector<Value> AppendToVector(const ArrayRef<Value> arguments, Value append) { SmallVector<Value> ret(arguments); ret.push_back(append); return ret; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 21.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/replace_stablehlo_ops_in_main_function_with_xla_call_module_ops.cc
const SetVector<Operation*>& ops_to_add, SmallVector<Operation*>& all_descendants) { if (!op) { return false; } SmallVector<Operation*> current_layer_descendants; SmallVector<Operation*> next_layer_descendants; int current_depth = 0; current_layer_descendants.push_back(op); // BFS downstream ops for current user.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 21K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/legalize_tensorlist.td
def Size1InputRange : NativeCodeCall< "SmallVector<Value, 1>{$0}">; def Size2InputRange : NativeCodeCall< "SmallVector<Value, 2>{$0, $1}">; def Size3InputRange : NativeCodeCall< "SmallVector<Value, 3>{$0, $1, $2}">; def EmptyCustomOptions : NativeCodeCall< "TFL::ConstBytesAttr::get($_builder.getContext(), \"\")">;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 18 07:12:51 UTC 2023 - 3.4K bytes - Viewed (0)