- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 43 for IfRegion (0.41 sec)
-
tensorflow/compiler/mlir/tensorflow/tests/resource-device-inference.mlir
-> !tf_res // CHECK-NEXT: "tf.VarHandleOp" %var_handle = "tf.VarHandleOp"() {container = "c", shared_name = "v0", device = "/TPU:1"} : () -> !tf_res // CHECK-NEXT: "tf.IfRegion" "tf.IfRegion"(%arg1) ({ func.call @ifregion_then(%id0, %var_handle) : (!tf_res, !tf_res) -> () "tf.Yield"() : () -> () }, {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 17 16:01:45 UTC 2022 - 18.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/shape_inference.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jan 23 17:24:10 UTC 2024 - 167.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/decompose_reduce_dataset.mlir
// CHECK: %[[GET_NEXT:[0-9]*]] = "tf.IteratorGetNextAsOptional"(%[[ANON_ITER]]) // CHECK-NEXT: %[[HAS_VALUE:[0-9]*]] = "tf.OptionalHasValue"(%[[GET_NEXT]]) // CHECK-NEXT: %[[IF:.*]] = "tf.IfRegion"(%[[HAS_VALUE]]) // CHECK-NEXT: %[[GET_VALUE:[0-9]*]] = "tf.OptionalGetValue"(%[[GET_NEXT]]) // CHECK-NEXT: %[[FUNC_CALL:[0-9]*]] = func.call @__reduce_func_1(%[[ARG_5]], %[[GET_VALUE]])
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Aug 18 17:16:34 UTC 2022 - 9.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tpu-cluster-cleanup-attributes.mlir
// CHECK-NOT: _replication_info = // CHECK-NOT: _xla_compile_device_type // CHECK-NOT: device = %1 = "tf_device.cluster"() ({ %2 = "tf.Add"(%arg1, %arg1) : (tensor<f32>, tensor<f32>) -> tensor<f32> %3 = "tf.IfRegion"(%arg0) ({ %4 = "tf.Mul" (%arg1, %2) {device = "y"}: (tensor<f32>, tensor<f32>) -> tensor<f32> "tf.Yield"(%4) : (tensor<f32>) -> () }, { // CHECK: device = "/device:TPU_REPLICATED_CORE:0"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 24 05:47:26 UTC 2022 - 2.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/resource_op_lifting.mlir
// CHECK: %[[IF:.*]]:2 = "tf.IfRegion"(%[[ARG0]]) %0 = "tf.VarHandleOp"() {container = "c", shared_name = "v"} : () -> tensor<*x!tf_type.resource<tensor<4xf32>>> %1 = "tf.VarHandleOp"() {container = "c", shared_name = "v2"} : () -> tensor<*x!tf_type.resource<tensor<4xf32>>> %2 = "tf_device.cluster"() ({ %3:2 = "tf.IfRegion"(%arg0) ({ // CHECK-NEXT: %[[CONST:.*]] = "tf.Const"()
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 74K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tf-ops.mlir
// ----- // Test invalid tf.Yield operation (parent should be IfRegion) func.func @testInvalidYieldOp(%arg0: f32) -> () { // expected-error @+1 {{'tf.Yield' op expects parent op to be one of 'tf.CaseRegion, tf.IfRegion, tf.WhileRegion, tf.GeneratorDatasetRegion'}} "tf.Yield"(%arg0) : (f32) -> () } // ----- // Test valid tf.IfRegion operation // CHECK-LABEL: func @testValidIfRegionOp
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 23 14:40:35 UTC 2023 - 236.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/tests/passes/insert_calibration_statistics_saver.mlir
return %1 : tensor<1x3xf32> } } // ----- // Check the IfRegion is set to stateful. module attributes {tf.versions = {bad_consumers = [], min_consumer = 12 : i32, producer = 1833 : i32}, tf_saved_model.semantics} { // CHECK-LABEL: func.func @serving_default // CHECK: "tf.IfRegion" // CHECK-SAME: is_stateless = false // CHECK: "tf.CalibrationStatisticsSaver"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 01:09:50 UTC 2024 - 24.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/analysis/resource_alias_analysis.h
llvm::ArrayRef<func::FuncOp> functions, const BacktrackAnalysis& backtrack_analysis); // Analyzes tf.CaseRegion/tf.IfRegion ops to compute resource IDs. void AnalyzeRegionCaseOrIfOp(Operation* case_or_if_op, const BacktrackAnalysis& backtrack_analysis);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 15 09:04:13 UTC 2024 - 7.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_passes.td
let summary = "Merges IfRegion ops together with a common predicate."; let description = [{ This pass merges IfRegion ops together if they have the same predicate and it is safe to do so (there are no intermediate dependencies, they are in the same block, etc). For example: ```mlir "tf.IfRegion"(%0) ( { %2 = "tf.A"() : () -> (tensor<f32>)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:18:05 UTC 2024 - 99.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tensor_array_ops_decomposition.mlir
// CHECK: "tf.IfRegion"(%[[PRED]]) <{is_stateless = false}> ({ %case_op = "tf.IfRegion"(%arg0) <{is_stateless = false}> ({ // CHECK: %[[TA_VAL:.*]] = "tf.ReadVariableOp"(%[[TA_BUFFER]]) // CHECK: "tf.Slice"(%[[TA_VAL]] // CHECK-NOT: tf.TensorArrayReadV3
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 49K bytes - Viewed (0)