- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 85 for partitioned (0.21 sec)
-
tensorflow/compiler/mlir/lite/tests/partitioned-topological-sort.mlir
// RUN: tf-opt %s -tfl-partitioned-topological-sort | FileCheck %s // CHECK-LABEL: @tf_ops_will_be_partitioned func.func @tf_ops_will_be_partitioned() -> tensor<1xf32> { %const = "tfl.pseudo_const"() {value = dense<[1.0]> : tensor<1xf32>} : () -> tensor<1xf32> %tmp1 = "tfl.add"(%const, %const) { fused_activation_function = "NONE" } : (tensor<1xf32>,tensor<1xf32>) -> (tensor<1xf32>)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Aug 19 22:33:49 UTC 2022 - 8.1K bytes - Viewed (0) -
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) -
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/tensorflow/tests/graphdef2mlir/empty-value-attr.pbtxt
} # This should be modeled as unit attribute attr { key: "_input_shapes" value { } } } } versions { producer: 85 min_consumer: 12 } # Match partitioned call in main and capture the callee name. # CHECK-LABEL: func @main # CHECK: tf.PartitionedCall # CHECK-SAME: f = @[[FUNCTION:[a-zA-Z0-9_]*]] # Verify that callee has the unit attribute tf._input_shapes.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Nov 11 19:14:04 UTC 2020 - 1.4K 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) -
tensorflow/compiler/mlir/tensorflow/tests/guarantee-all-funcs-one-use.mlir
func.func private @f() { func.call @g() : () -> () func.return } func.func private @g() { func.call @f() : () -> () func.return } } // ----- // Test stateful and stateless partitioned calls. // CHECK-LABEL: func @f func.func @f() { // CHECK: "tf.PartitionedCall"() <{config = "", config_proto = "", executor_type = "", f = @g}> : () -> ()
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 2.1K bytes - Viewed (0)