- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 345 for SmallVector (0.14 sec)
-
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 =
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/lite/utils/perception_ops_utils_test.cc
func::FuncOp createMaxUnpoolingFunc( mlir::Builder* builder, const SmallVector<int64_t, 4>& input_shape, const SmallVector<int64_t, 4>& output_shape) { auto input_type = RankedTensorType::get(input_shape, builder->getF32Type()); auto indices_type = RankedTensorType::get(input_shape, builder->getI64Type()); auto output_type = RankedTensorType::get(output_shape, builder->getF32Type()); SmallVector<mlir::Type, 2> input_types{input_type, indices_type};
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Sep 29 21:02:21 UTC 2022 - 7.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/utils/tf_to_xla_attribute_utils_test.cc
namespace mlir::quant { namespace { void PackOperandTestHelper( const llvm::SmallVector<int64_t>& unpacked_shape, const llvm::SmallVector<int8_t>& unpacked_values, int pack_dim, const llvm::SmallVector<int64_t>& expected_packed_shape, const llvm::SmallVector<int8_t>& expected_packed_values) { MLIRContext context; OwningOpRef<ModuleOp> module(ModuleOp::create(UnknownLoc::get(&context)));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 3.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/cc/convert_asset_args.cc
// of the index path attribute. SmallVector<NamedAttribute> ReplaceBoundInputAttrWithIndexPathAttr( const ArrayRef<NamedAttribute> arg_attrs, const StringRef index_path, Builder& builder) { // Keep all other attributes except the tf_saved_model.bound_input attribute, // as we are replacing it with tf_saved_model.index_path. SmallVector<NamedAttribute> new_arg_attrs; for (auto arg_attr : arg_attrs) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/lstm_utils.cc
// time-majored. shrink_axis_mask: should be 0b010 = 2 or 0b001 = 1 if it's // time-majored. SmallVector<int64_t, 2> last_output_shape({batch, n_output}); SmallVector<int32_t, 3> end({0, 0, 0}); SmallVector<int32_t, 3> strides({1, 1, 1}); SmallVector<int32_t, 3> begin; int64_t new_axis_mask = 0; int64_t ellipsis_mask = 0; int64_t begin_mask; int64_t end_mask;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 36.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tfg-to-tfe.cc
.getValue(), ftype); func->setAttrs(graph_func->getAttrs()); llvm::SmallVector<Type> arg_types; llvm::SmallVector<Type> res_types; llvm::SmallVector<DictionaryAttr> arg_attrs; llvm::SmallVector<DictionaryAttr> res_attrs; if (failed(FilterTfgSpecificArgResultAttributes( getContext(), ftype.getInputs(), graph_func.getAllArgAttrs(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 21.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/uniform_quantized_types_test.cc
const UniformQuantizedPerAxisType quantized_type = CreateI8F32UniformQuantizedPerAxisType( UnknownLoc::get(&ctx_), ctx_, /*scales=*/SmallVector<double, 2>{1.0, 1.0}, /*zero_points=*/SmallVector<int64_t, 2>{0, 0}, /*quantization_dimension=*/0); // Storage type of `i8` is currently verifiable as `unsigned` in `Types.cpp`.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 28.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/hoist_replicate_invariant_resource_writes.cc
SmallVector<TF::AssignVariableOp> GetTailWritesToReplicateInvariantResourceVars( tf_device::ReplicateOp replicate_op) { SmallVector<TF::AssignVariableOp, 16> tail_assign_variable_ops; llvm::SmallDenseSet<Value, 16> visited_resources; for (auto& op : llvm::reverse(replicate_op.getRegion().front().getOperations())) { SmallVector<Value> op_accessed_resources = GetAccessedResources(op);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Nov 03 12:35:38 UTC 2022 - 5.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/merge_control_flow.cc
} // Return a vector of the return indices. llvm::SmallVector<llvm::SmallVector<int, 4>> GetReturnIndicesVec( const llvm::SmallVector<TF::IfRegionOp, 8>& if_op_segment) { llvm::SmallVector<llvm::SmallVector<int, 4>> return_indices_vec; for (auto it = if_op_segment.begin(); it != if_op_segment.end(); ++it) { llvm::SmallVector<int, 4> indices_to_keep_vec = GetReturnIndicesToKeep(*it, if_op_segment);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 17 07:31:01 UTC 2023 - 25.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/extract_outside_compilation.cc
device_cluster.getLoc(), num_regions, llvm::ArrayRef<Type>{}); SmallVector<Operation*, 4> core_to_host_insertion_point; SmallVector<mlir::tf_device::LaunchOp, 4> core_to_tmp_launch; SmallVector<Operation*, 4> compilation_key_ops; SmallVector<Value, 4> core_to_compilation_key; SmallVector<Operation*, 4> core_to_device_ordinal_op; SmallVector<Value, 4> core_to_device_ordinal;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 21:25:12 UTC 2024 - 68.3K bytes - Viewed (0)