- Sort Score
- Result 10 results
- Languages All
Results 31 - 40 of 66 for SmallVectorImpl (0.24 sec)
-
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_a_m.cc
// s0_shape & s1_shape. bool ExtractInputConstShape(BroadcastGradientArgsOp op, DenseIntElementsAttr& s0, DenseIntElementsAttr& s1, SmallVectorImpl<int64_t>& s0_shape, SmallVectorImpl<int64_t>& s1_shape) { if (!matchPattern(op.getS0(), m_Constant(&s0))) return false; if (!matchPattern(op.getS1(), m_Constant(&s1))) return false;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 146.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf_communication.cc
// the parent token. Also we use the same implicit token for use in the If op's // regions. void RewriteRegionIfOp(OpBuilder& builder, IfOp region_if, SmallVectorImpl<OpVisitorState>& ops_to_visit, Value token) { llvm::SmallDenseMap<Value, Value> rewritten_operands; auto new_result_types =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 40.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo_conversions/dot_general.cc
template <typename ValueT, typename Range> void Append(llvm::SmallVectorImpl<ValueT>& values, Range&& range) { values.insert(values.end(), range.begin(), range.end()); } // Appends all elements in `range` to `values`. template <typename ValueT, typename Range, typename... RangeTs> void Append(llvm::SmallVectorImpl<ValueT>& values, Range&& range, RangeTs&&... ranges) {
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/lite/quantization/device_target.h
#include "tensorflow/compiler/mlir/lite/quantization/numerical_utils.h" namespace mlir { namespace quant { class QuantizeContext; using AdjacentOperations = llvm::SmallVectorImpl<Operation*>; using QuantizedMultipliers = llvm::SmallVector<QuantizedMultiplier, 4>; using QuantizedRanges = llvm::SmallVector<QuantizedRange, 4>; using ScaleFn = std::function<LogicalResult(QuantizeContext*, Operation*,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 08 10:41:08 UTC 2024 - 7.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/cluster_formation.cc
continue; use.set(std::get<1>(p)); } } } // Get all escaped live-out values of a region. void GetLiveOuts(Region* region, llvm::SmallVectorImpl<Value>* live_outs) { live_outs->clear(); for (Operation& op : region->front()) { for (Value v : op.getResults()) { // A value is live-out if any of its users are not inside value producer's
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Oct 05 13:30:21 UTC 2023 - 6.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_utils.h
// scales. quant::QuantizedType DownCastScale(quant::QuantizedType type, const SmallVectorImpl<double>& mins, const SmallVectorImpl<double>& maxs, Location loc); quant::QuantizedType DownCastScale(quant::QuantizedType type, double min,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 20:30:06 UTC 2024 - 41.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tfg-to-tfe.cc
} return mlir::success(); } static mlir::LogicalResult ReformatOpAttributes( mlir::MLIRContext *context, llvm::ArrayRef<mlir::NamedAttribute> attrs, llvm::SmallVectorImpl<mlir::NamedAttribute> &output) { for (mlir::NamedAttribute attr : attrs) { if (attr.getName().strref().contains( mlir::tfg::TFGraphDialect::getDeviceAttrKey())) {
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/tf2xla/internal/passes/extract_head_tail_outside_compilation.cc
void FindOutsideCompiledOpsAtTailAndClusterResults( const mlir::TF::SideEffectAnalysis& side_effect_analysis, mlir::tf_device::ClusterOp cluster, llvm::SmallVectorImpl<Operation*>* tail_outside_compiled_ops, llvm::SmallVectorImpl<Value>* cluster_results) { const auto& analysis = side_effect_analysis.GetAnalysisForFunc( cluster->getParentOfType<FuncOp>()); Region* cluster_region = &cluster.getBody();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 20.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/perception_ops_utils.cc
} } return success(); } inline LogicalResult GetIntegerArraySafe( func::FuncOp* func, const DictionaryAttr& attrs, const std::string& attr_name, llvm::SmallVectorImpl<int32_t>* results, int N) { ArrayAttr array_attr = mlir::dyn_cast_or_null<ArrayAttr>(attrs.get(attr_name)); if (array_attr == nullptr || array_attr.size() != N) { return func->emitError()
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 8.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/converter_gen.cc
// Signature: // void mlir::BuiltinOptions{id}ToAttributes( // tflite::BuiltinOptions{id}Union op_union, // mlir::Builder builder, // llvm::SmallVectorImpl<mlir::NamedAttribute> &attributes); // // where id is an empty string if builtin_options_id is 1, or builtin_options_id // otherwise. static void EmitBuiltinOptionsToAttributes(const RecordKeeper &record_keeper,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Dec 19 15:05:28 UTC 2023 - 23.7K bytes - Viewed (0)