- Sort Score
- Result 10 results
- Languages All
Results 31 - 40 of 345 for SmallVector (0.23 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/xla_validate_inputs.cc
void runOnOperation() override; }; LogicalResult HasNoNestedEntryFunctions( const llvm::SmallVector<func::FuncOp> &entry_funcs, SymbolTable &symtab) { auto calls_entry_functions = [&](SymbolUserOpInterface op) { llvm::SmallVector<func::FuncOp> callees; if (GetCallees(op, symtab, callees).failed()) { return false; } for (auto &callee : callees) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 06 19:29:14 UTC 2024 - 3.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/tpu_rewrite_device_util.cc
// every host, a failure will be returned. absl::StatusOr<llvm::SmallVector<llvm::SmallVector<ParsedDevice, 8>, 8>> GetTPUDevices(ParsedDevices devices, llvm::ArrayRef<ParsedDevice> system_devices) { llvm::SmallVector<llvm::SmallVector<ParsedDevice, 8>, 8> tpu_devices; tpu_devices.reserve(system_devices.size());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Jun 10 20:10:40 UTC 2024 - 32.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/common/outline_operations.h
// BlockArg or as a result of an Operation. llvm::SmallVector<Value> AccumulateOperandsDefinedAbove( const llvm::SetVector<Operation*>& partition_ops); // Similar to `AccumulateOperandsDefinedAbove()`, computes the Value(s) that are // defined within a Subgraph and referenced in a descendant Operation. These // Values(s) are to be returned by the new raised function. llvm::SmallVector<Value> AccumulateResultsDefinedWithin(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Nov 17 18:49:43 UTC 2022 - 6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/xla_sharding_util.cc
} return mlir::success(); } llvm::SmallVector<llvm::SmallVector<int64_t, 4>, 4> GetMetadataArgumentMapping( const tpu::TPUCompileMetadataProto& metadata) { llvm::SmallVector<llvm::SmallVector<int64_t, 4>, 4> input_mappings( metadata.num_cores_per_replica(), llvm::SmallVector<int64_t, 4>()); if (metadata.num_cores_per_replica() == 1) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 22 21:28:13 UTC 2024 - 34K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/utils.h
SmallVector<int64_t, 8> perm_values; for (const auto& dim : perm_values_attr.getValues<APInt>()) perm_values.push_back(dim.getSExtValue()); // This should never happen unless the input graph is malformed. if (input_shape.size() != perm_values.size()) { return false; } SmallVector<int, 8> old_major_index_ordering; SmallVector<int, 8> new_major_index_ordering;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 00:40:15 UTC 2024 - 11.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_parallel_execute_sink_resource_write.cc
bool rewrite = false; const int num_regions = parallel_execute.getNumRegions(); llvm::SmallVector<Value, 4> results_to_remap; // Go through each region and find AssignVariableOps that can be moved into // the parallel_execute region. Result indices by region index are collected, // so they can be removed afterwards. llvm::SmallVector<llvm::SmallVector<int, 4>, 4> results_to_remove_by_region; results_to_remove_by_region.resize(num_regions);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Dec 06 04:46:18 UTC 2022 - 6.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/replace_cast_hacks_with_tf_xla_ops.cc
Value &window_strides, Value &lhs_dilation, Value &rhs_dilation, Value &feature_group_count, int num_dims) { SmallVector<int32_t> lhs_dilation_values(num_dims - 2, 1); SmallVector<int32_t> stride_values, rhs_dilation_values; for (int64_t i : llvm::seq<int64_t>(1, num_dims - 1)) { stride_values.push_back(mlir::cast<IntegerAttr>(strides[i]).getInt());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 47.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/prepare_tf.cc
return failure(); } SmallVector<int32_t, 4> begin, end, strides; SmallVector<int32_t, 4> padded_begin, padded_end, padded_strides; int num_input_dims = ranked_input_type.getRank(); SmallVector<int32_t, 4> padding_begin(num_input_dims, 0); auto input_shape = ranked_input_type.getShape(); SmallVector<int32_t, 4> padding_end(input_shape.begin(), input_shape.end());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:49:50 UTC 2024 - 64.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/analysis/side_effect_analysis.h
// pass the filter (returning true) will be included. const llvm::SmallVector<Operation*, 4>& DirectControlPredecessors( Operation* op) const; llvm::SmallVector<Operation*, 4> DirectControlPredecessors( Operation* op, llvm::function_ref<bool(Operation*)> filter) const; // pass the filter (returning true) will be included. const llvm::SmallVector<Operation*, 4>& DirectControlSuccessors( Operation* op) const;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 15 09:04:13 UTC 2024 - 14.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/call_graph_util.h
bool IsEntryFunction(func::FuncOp func); // Get all the entry functions in an MLIR module. llvm::SmallVector<func::FuncOp> GetEntryFunctions(ModuleOp module); // Get all the functions referenced in a symber user op and save them in // `callees`. LogicalResult GetCallees(SymbolUserOpInterface op, SymbolTable &symtab, llvm::SmallVector<func::FuncOp> &callees);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Dec 16 06:18:49 UTC 2023 - 4.3K bytes - Viewed (0)