- Sort Score
- Result 10 results
- Languages All
Results 71 - 80 of 147 for SmallVector (0.24 sec)
-
tensorflow/compiler/mlir/tensorflow/utils/tf_xla_mlir_translate.cc
TF_RETURN_IF_ERROR( tensorflow::ParseNodeShapes(input_shapes_str, input_shapes_vector)); llvm::SmallVector<DataType, 4> dtypes_vector; TF_RETURN_IF_ERROR(ParseDataTypes(input_dtypes_str, dtypes_vector)); llvm::SmallVector<XlaArgument::Kind, 4> arg_kinds_vector; TF_RETURN_IF_ERROR(ParseArgumentKinds(arg_kinds_str, arg_kinds_vector)); if (input_shapes_vector.empty())
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 18.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/hlo_matchers.cc
// If there is no valid next index, returns `std::nullopt`. // // `index` should have the same size as `shape`. // Each value `dim` in `index` should be in [0, shape[dim]). static std::optional<SmallVector<int64_t>> NextTensorIndex( SmallVector<int64_t> index, ArrayRef<int64_t> shape, int64_t fixed_axis) { #ifndef NDEBUG assert(shape.size() == index.size()); assert(fixed_axis < shape.size()); assert(fixed_axis >= 0);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/tf2xla_rewriter.cc
#include <vector> #include "absl/container/inlined_vector.h" #include "absl/memory/memory.h" #include "absl/strings/string_view.h" #include "llvm/ADT/DenseSet.h" #include "llvm/ADT/STLExtras.h" #include "llvm/ADT/SmallVector.h" #include "llvm/Support/Casting.h" #include "llvm/Support/raw_ostream.h" #include "mlir/Dialect/Func/IR/FuncOps.h" // from @llvm-project #include "mlir/Dialect/SparseTensor/IR/SparseTensor.h" // from @llvm-project
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:16:07 UTC 2024 - 18.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tensor_array_ops_decomposition.cc
// partitioned call ops. struct PartitionedCallTensorArrayOpsInfo { bool signature_change; func::FuncOp decomposed_callee; llvm::SmallVector<std::pair<int64_t, llvm::SmallVector<string, 4>>, 4> arg_grads; llvm::SmallVector<std::pair<int64_t, int64_t>, 4> ret_forward_input; }; // Updates a called function's input signature by adjusting resource types, and // adding required gradient arguments.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 02 20:41:19 UTC 2023 - 40.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/analysis/resource_alias_analysis.cc
#include "llvm/ADT/ArrayRef.h" #include "llvm/ADT/DenseMap.h" #include "llvm/ADT/SCCIterator.h" #include "llvm/ADT/STLExtras.h" #include "llvm/ADT/SetVector.h" #include "llvm/ADT/SmallSet.h" #include "llvm/ADT/SmallVector.h" #include "llvm/ADT/iterator_range.h" #include "llvm/Support/Casting.h" #include "mlir/Analysis/CallGraph.h" // from @llvm-project #include "mlir/Dialect/Func/IR/FuncOps.h" // from @llvm-project
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 15 09:04:13 UTC 2024 - 28.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/region_control_flow_to_functional.cc
LogicalResult RegionControlFlowToFunctional::ConvertCaseOp( SymbolTableCollection& symbol_table, CaseRegionOp case_region) { llvm::SmallVector<mlir::Attribute, 2> branch_symbols; // Collect external values that are used within all branches. llvm::SmallVector<Value, 4> extern_values = CollectExternValues(case_region.getRegions()); // These external values need to be added as inputs to the generated Case. The
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 28.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/prepare_quantize_helper.h
// the reduce operation after element-wise multiplication between state and // time weights. if (tensor_property.number_of_bits == 10) { SmallVector<double, 4> mins(1, std::numeric_limits<double>::max()); SmallVector<double, 4> maxs(1, std::numeric_limits<double>::min()); // Computes the effective min/max values of the attribute values.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 18:01:23 UTC 2024 - 28K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/quantization/import_quant_stats_pass.cc
limitations under the License. ==============================================================================*/ #include <memory> #include "absl/strings/str_split.h" #include "llvm/ADT/APFloat.h" #include "llvm/ADT/SmallVector.h" #include "llvm/ADT/StringMap.h" #include "llvm/Support/CommandLine.h" #include "llvm/Support/Regex.h" #include "mlir/Dialect/Func/IR/FuncOps.h" // from @llvm-project
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 08 10:41:08 UTC 2024 - 9.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/unfuse_batch_norm_pass.cc
limitations under the License. ==============================================================================*/ #include <cstdint> #include <memory> #include <utility> #include "llvm/ADT/SmallVector.h" #include "mlir/Dialect/Func/IR/FuncOps.h" // from @llvm-project #include "mlir/Dialect/Shape/IR/Shape.h" // from @llvm-project #include "mlir/IR/Attributes.h" // from @llvm-project
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/xla_call_module_deserialization.cc
#include <memory> #include <string> #include <utility> #include <vector> #include "llvm/ADT/ArrayRef.h" #include "llvm/ADT/STLExtras.h" #include "llvm/ADT/SmallVector.h" #include "llvm/ADT/StringRef.h" #include "mlir/Dialect/Func/Extensions/AllExtensions.h" // from @llvm-project #include "mlir/Dialect/Func/IR/FuncOps.h" // from @llvm-project
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 23 09:05:47 UTC 2024 - 11.1K bytes - Viewed (0)