Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for StatefulPartitionedCall (0.24 sec)

  1. tensorflow/compiler/mlir/lite/flatbuffer_export.cc

      return false;
    }
    
    // Returns whether the current op is not supported by the TF Lite runtime.
    static bool IsUnsupportedFlexOp(const std::string& op_name) {
      return op_name == "PartitionedCall" || op_name == "StatefulPartitionedCall";
    }
    
    // Create description of operation that could not be converted.
    static std::string GetOpDescriptionForDebug(Operation* inst) {
      const int kLargeElementsAttr = 16;
      std::string op_str;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 21:41:49 UTC 2024
    - 164.5K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/tests/canonicalize.mlir

      %2 = "tf.If"(%0, %arg0, %arg1) {then_branch = @add, else_branch = @sub, output_shapes = [#tf_type.shape<>], device = "noodle", is_stateless = true} : (tensor<i1>, tensor<f32>, tensor<f32>) -> tensor<f32>
      // CHECK: %1 = "tf.StatefulPartitionedCall"(%0, %arg1)
      // CHECK-SAME: f = @add
      // CHECK-SAME: _underscore_attr = "something"
      // CHECK-SAME: device = "noodle"
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 09 22:07:10 UTC 2024
    - 132.1K bytes
    - Viewed (0)
Back to top