- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for control_node (0.17 sec)
-
tensorflow/compiler/mlir/lite/tests/flatbuffer2mlir/control_edges.mlir
%tmp2, %ctl2 = tfl.control_node(%ctl1) controls "tfl.neg"(%tmp1): (tensor<1xf32>) -> tensor<1xf32> %tmp3, %ctl3 = tfl.control_node(%ctl2) controls "tfl.neg"(%tmp2): (tensor<1xf32>) -> tensor<1xf32> func.return %tmp3: tensor<1xf32> } // CHECK-NEXT: %[[t0:.*]], %[[c0:.*]] = tfl.control_node controls "tfl.neg"(%arg0)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Oct 14 21:40:53 UTC 2022 - 3.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/pin-ops-with-side-effects.mlir
} // CHECK-NEXT: %[[VAR:.*]], %[[CONTROL:.*]] = tfl.control_node controls "tfl.var_handle" // CHECK-NEXT: return %[[VAR]] // CHECK-LABEL: @tfl_call_once_gets_control_node func.func @tfl_call_once_gets_control_node() -> () { "tfl.call_once"() { session_init_function = "noop" } : () -> () func.return } // CHECK-NEXT: control_node controls "tfl.call_once" // CHECK-NEXT: return
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Aug 17 10:45:19 UTC 2022 - 5.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/mlir2flatbuffer/simple_with_connected_control_nodes.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Aug 17 13:32:53 UTC 2022 - 4.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/mlir2flatbuffer/simple_with_unconnected_control_nodes.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Aug 17 13:32:53 UTC 2022 - 4.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/tests/legalize-stablehlo-vhlo.mlir
func.func @op_with_tfl_control_flow() -> (tensor<1xf32>, !tfl.control) { // CHECK: vhlo.constant_v1 %0 = stablehlo.constant dense<1.000000e+00> : tensor<1xf32> // CHECK-NEXT: tfl.control_node %outputs, %control = tfl.control_node { %1 = "tfl.neg"(%0) : (tensor<1xf32>) -> tensor<1xf32> "tfl.yield"(%1) : (tensor<1xf32>) -> () } return %outputs, %control : tensor<1xf32>, !tfl.control } // -----
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 07 22:39:35 UTC 2024 - 5.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/partitioned_topological_sort.cc
#include "tensorflow/compiler/mlir/lite/transforms/passes.h.inc" bool IsFlexDelegate(Operation *op) { // Unwrap from any ControlNodeOps or CustomTfOp. if (auto control_node = dyn_cast<mlir::TFL::ControlNodeOp>(*op)) { return IsFlexDelegate(&control_node.getBody().front().front()); } if (auto custom_tf_op = dyn_cast<mlir::TFL::CustomTfOp>(*op)) { return IsFlexDelegate(&custom_tf_op.getBody().front().front()); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Dec 19 15:05:28 UTC 2023 - 7.2K bytes - Viewed (0)