- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 325 for partitioned (0.39 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/tests/legalize-skip-partitioned-calls.mlir
// RUN: odml-to-stablehlo-opt %s --tf-stablehlo=skip-partitioned-calls=true | FileCheck %s --check-prefix=CHECK-SKIP // RUN: odml-to-stablehlo-opt %s --tf-stablehlo=skip-partitioned-calls=false | FileCheck %s --check-prefix=CHECK-NOSKIP module { func.func @partitioned_call(%arg0: tensor<1x2x2x3xf32>) -> (tensor<1x2x2x3xf32>) { %0 = "tf.StatefulPartitionedCall"(%arg0) <{ config = "", config_proto = "", executor_type = "", f = @some_func }> {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 08 20:05:12 UTC 2024 - 1.5K bytes - Viewed (0) -
doc/next/6-stdlib/99-minor/net/http/62490.md
The new [Cookie.Partitioned] field identifies cookies with the Partitioned attribute....
Registered: Wed Jun 12 16:32:35 UTC 2024 - Last Modified: Wed May 22 18:33:05 UTC 2024 - 85 bytes - Viewed (0) -
tensorflow/c/checkpoint_reader.h
// // The class currently only interacts with single-slice (i.e., non-partitioned) // variables. class CheckpointReader { public: CheckpointReader(const string& filename, TF_Status* status); bool HasTensor(const string& name) const; const string DebugString() const; // Returns a map from variable names to their shapes. Slices of a partitioned // tensor are combined into a single entry.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Oct 12 08:49:52 UTC 2023 - 3.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/xla_cluster_formation.mlir
func.func @stateful_pcall_func(%arg0: tensor<i32>) -> tensor<i32> { func.return %arg0 : tensor<i32> } // ----- // Check that we only outline the outermost partitioned call with // `_xla_compile_device_type` in nested calls to a device cluster. The callee // of any partitioned call outside of a device cluster is marked with // `tf._noinline = true` to prevent it getting inlined to perseve runtime // performance.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Dec 19 19:09:44 UTC 2023 - 4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/xla_cluster_formation.cc
using mlir::SymbolUserOpInterface; using mlir::func::FuncOp; #define GEN_PASS_DEF_XLACLUSTERFORMATIONPASS #include "tensorflow/compiler/mlir/tf2xla/internal/passes/clustering_passes.h.inc" // Outlines partitioned call ops with `_XlaMustCompile` to device clusters. struct XlaClusterFormationPass : public impl::XlaClusterFormationPassBase<XlaClusterFormationPass> { void runOnOperation() override; };
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Dec 19 19:09:44 UTC 2023 - 6K bytes - Viewed (0) -
src/net/http/cookie.go
Registered: Wed Jun 12 16:32:35 UTC 2024 - Last Modified: Wed May 22 18:33:05 UTC 2024 - 13.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_resource_partitioning.cc
} } // namespace // Rewrites unpartitioned resource reads and writes to partitioned resource // reads and writes. The TPU computation from the frontend is generated in such // a way that resource operations operate on the unpartitioned resource handle // (from a `tf.TPUReplicatedInput`). This results in resource reads and writes // on the unpartitioned resource handle post resource op decomposition/lifting.
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/tensorflow/tests/end-to-end-tpu-reshard-variables.mlir
"tf.StatefulPartitionedCall"(%arg0) <{config = "", config_proto = "", executor_type = "", f = @partitioned}> : (tensor<*x!tf_type.resource>) -> () tf_executor.yield } tf_executor.fetch %control : !tf_executor.control } return }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 13 21:23:47 UTC 2024 - 4.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/tpu_cluster_formation.cc
using PartitionedClusterOutputMap = absl::flat_hash_map< uint64_t, llvm::SmallVector<mlir::TF::TPUPartitionedOutputV2Op, 8>>; // Returns the partitioned output ops from the cluster if there are any, // along with any single user identity ops between them. Not all outputs // of a cluster must be partitioned, so the output is a map from cluster // output ids to ops. std::tuple<PartitionedClusterOutputMap,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 22:03:30 UTC 2024 - 39.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/tf_stablehlo_pass.cc
Option<bool> skip_resize_{ *this, "skip-resize", ::llvm::cl::desc("Skip tf.ResizeBilinear and tf.ResizeNearestNeighbor")}; Option<bool> skip_partitioned_calls_{ *this, "skip-partitioned-calls", ::llvm::cl::desc( "Skip tf.StatefulPartitionedCall and tf.PartitionedCall")}; }; void TFToMhloPass::runOnOperation() { auto func = getOperation();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:49:50 UTC 2024 - 7.5K bytes - Viewed (0)