- Sort Score
- Result 10 results
- Languages All
Results 111 - 120 of 345 for SmallVector (0.21 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/tpu_resource_partitioning.cc
==============================================================================*/ #include <memory> #include <tuple> #include "llvm/ADT/MapVector.h" #include "llvm/ADT/STLExtras.h" #include "llvm/ADT/SmallVector.h" #include "llvm/Support/Casting.h" #include "mlir/Dialect/Func/IR/FuncOps.h" // from @llvm-project #include "mlir/IR/BuiltinOps.h" // from @llvm-project #include "mlir/IR/BuiltinTypes.h" // from @llvm-project
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 11.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/runtime_fallback/runtime_fallback_executor.h
#define TENSORFLOW_COMPILER_MLIR_TFRT_RUNTIME_FALLBACK_RUNTIME_FALLBACK_EXECUTOR_H_ #include <memory> #include "llvm/ADT/ArrayRef.h" #include "llvm/ADT/SmallVector.h" #include "llvm/ADT/StringRef.h" #include "tensorflow/core/framework/tensor.h" #include "tensorflow/core/platform/threadpool_interface.h" #include "tfrt/bef/bef_buffer.h" // from @tf_runtime
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Nov 14 22:40:40 UTC 2022 - 2.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/update_control_dependencies.cc
int& num_control_inputs_removed, int& num_control_inputs_added) { llvm::SmallVector<Operation*, 4> control_deps = analysis_for_func.DirectControlPredecessors(op); ClearControlInputs(op, num_control_inputs_removed); llvm::SmallVector<Operation*, 8> preds_in_reverse_program_order( control_deps.begin(), control_deps.end());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jan 30 07:53:51 UTC 2024 - 8.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/insert_restore_op.cc
// `tf.AssignVariableOp`s. void CreateRestoreV2Op(std::vector<TF::VarHandleOp>& target_var_handle_ops, func::FuncOp session_init_func) { SmallVector<Type> tensor_types{}; SmallVector<std::string> tensor_names{}; for (auto var_handle_op : target_var_handle_ops) { tensor_names.emplace_back(var_handle_op.getSharedName().str());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Mar 12 06:02:20 UTC 2023 - 9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_layout_helper.h
#include "mlir/IR/Types.h" // from @llvm-project #include "tensorflow/compiler/mlir/tensorflow/ir/tf_op_interfaces.h" namespace mlir { class MLIRContext; namespace TF { SmallVector<int64_t, 4> ReversePermutation(ArrayRef<int64_t> permutation); SmallVector<int64_t, 4> GetDataFormatPermutation(StringRef from, StringRef to); // Shuffle elements in the `attr` according to the permutation. Optional
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 08 01:19:25 UTC 2023 - 5.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_saved_model_test.cc
"tf_saved_model.session_initializer"() {initializers = []} : () -> () } )mlir"; Block block; ModuleOp module_op = ParseModuleOp(kModuleOpStr, block, ctx_); SmallVector<func::FuncOp, 2> init_func_ops = GetInitializerFunctions(module_op); EXPECT_THAT(init_func_ops, IsEmpty()); } TEST_F(TfSavedModelTest,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 01 05:03:09 UTC 2023 - 6.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/tpu_rewrite_device_util.h
// their associated host CPU devices (for outside compilation). They are ordered // by `num_replicas` followed by `num_cores_per_replica`. using TPUDevicesAndHosts = llvm::SmallVector<llvm::SmallVector<TPUDeviceAndHost, 8>, 8>; // TPU compilation device, execution and associated host devices, and optionally // execution device IDs. Execution device IDs are populated if `topology` and
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Apr 26 09:37:10 UTC 2024 - 11.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/replicate_to_island.cc
// Collect result types and operands. Operation& terminator = replicate_op.GetBody().back(); llvm::SmallVector<Type, 8> output_types(terminator.getOperandTypes()); auto control_type = tf_executor::ControlType::get(island_op.getContext()); llvm::SmallVector<Value, 8> replica_inputs(island_op.getControlInputs()); // Replace replicate terminator with YieldOp. builder.setInsertionPoint(&terminator);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Jul 24 21:01:40 UTC 2023 - 16.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/breakup-islands.cc
} } OpBuilder builder(func); // For every op, add new control inputs in reverse order so that the ops don't // get invalidated. llvm::SmallVector<Value, 8> operands; llvm::SmallPtrSet<Operation*, 4> defining_ops; llvm::SmallVector<Type, 4> types; for (auto& item : llvm::make_early_inc_range(llvm::reverse(graph_op.GetBody()))) { auto it = new_control_inputs.find(&item);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Aug 11 20:52:36 UTC 2023 - 16.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/quantization/device_target.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*, AdjacentOperations*, bool*)>;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 08 10:41:08 UTC 2024 - 7.1K bytes - Viewed (0)