- Sort Score
- Result 10 results
- Languages All
Results 21 - 30 of 178 for SmallVector (0.14 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/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/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) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/insert_calibration_statistics_saver.cc
void FindCustomAggregatorOps( Region& region, const std::unordered_set<std::string>& aggregator_ops_to_ignore, SmallVector<Value>& statistics_outputs, SmallVector<StringRef>& ids, SmallVector<int32_t>& calibration_methods) { for (auto op : region.getOps<TF::CustomAggregatorOp>()) { if (aggregator_ops_to_ignore.count(op.getId().str())) continue;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 14 06:31:57 UTC 2024 - 7.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/lift_as_function_call.h
#define TENSORFLOW_COMPILER_MLIR_QUANTIZATION_COMMON_LIFT_AS_FUNCTION_CALL_H_ #include "absl/base/nullability.h" #include "absl/status/statusor.h" #include "llvm/ADT/SmallVector.h" #include "mlir/Bytecode/BytecodeOpInterface.h" // from @llvm-project #include "mlir/Dialect/Func/IR/FuncOps.h" // from @llvm-project #include "mlir/IR/Attributes.h" // from @llvm-project
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 10 04:07:09 UTC 2024 - 5.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/xla_inline_device_ops.cc
limitations under the License. ==============================================================================*/ // This pass remove Cluster ops by inlining Cluster ops. #include "llvm/ADT/SmallVector.h" #include "tensorflow/compiler/mlir/tensorflow/ir/tf_device.h" #include "tensorflow/compiler/mlir/tensorflow/ir/tf_ops.h" #include "tensorflow/compiler/mlir/tensorflow/ir/tf_types.h"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 04 00:59:46 UTC 2022 - 2.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_identity_pruning.cc
if (func && reachable_funcs.insert(func).second) funcs_to_visit.push_back(func); }); }; SmallVector<func::FuncOp, 4> funcs_to_visit; collect_reachable_funcs(region, funcs_to_visit); while (!funcs_to_visit.empty()) { SmallVector<func::FuncOp, 4> new_funcs_to_visit; for (func::FuncOp func_to_visit : funcs_to_visit) { if (!func_to_visit.getCallableRegion()) continue;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 05 23:50:19 UTC 2022 - 4.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/optimize_layout.cc
ArrayRef<int64_t> transpose_perm = transpose_op.getPermutation(); SmallVector<int64_t> new_padding_low = InvertPermutationToVector(pad_op.getEdgePaddingLow(), transpose_perm); SmallVector<int64_t> new_padding_high = InvertPermutationToVector(pad_op.getEdgePaddingHigh(), transpose_perm); SmallVector<int64_t> new_padding_interrier =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 21:59:06 UTC 2024 - 8.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_op_interfaces.td
return $_op.getDataFormat(); }]>, InterfaceMethod< [{Returns indices of layout dependent arguments.}], "SmallVector<unsigned, 4>", "GetLayoutDependentArgs", (ins) >, InterfaceMethod< [{Returns indices of layout dependent results.}], "SmallVector<unsigned, 4>", "GetLayoutDependentResults", (ins) >, InterfaceMethod<
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Nov 30 19:07:07 UTC 2022 - 6.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/dynamic_shape_utils.h
#define TENSORFLOW_COMPILER_MLIR_TENSORFLOW_UTILS_DYNAMIC_SHAPE_UTILS_H_ #include "mlir/IR/BuiltinTypes.h" // from @llvm-project namespace tensorflow { llvm::SmallVector<int64_t> ConvertTFShapeToMlir(llvm::ArrayRef<int64_t> shapes); llvm::SmallVector<int64_t> ConvertMlirShapeToTF(llvm::ArrayRef<int64_t> shape); static constexpr int64_t kTFDynamicSize = -1;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 18 11:40:17 UTC 2022 - 1.4K bytes - Viewed (0)