Search Options

Results per page
Sort
Preferred Languages
Advance

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

  1. tensorflow/compiler/mlir/tfrt/tests/ifrt/sink_variable_as_named_array.mlir

    //
    // CHECK-LABEL:  func.func @serving_default
    // CHECK-NOT:  IfrtLoadVariable
    // CHECK:      "tf.VarHandleOp"
    // CHECK-NEXT: "tf.AssignVariableOp"
    // CHECK-NEXT: "tf.ReadVariableOp"
    // CHECK-NEXT: "tf.StatefulPartitionedCall"
    // CHECK-NEXT:  return 
    //
    module {
      func.func @serving_default() -> tensor<*xi32> {
        %cst = "tf.Const"() <{value = dense<"some_test.txt"> : tensor<!tf_type.string>}> : () -> tensor<!tf_type.string>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 06 15:33:17 UTC 2024
    - 5.3K bytes
    - Viewed (0)
  2. 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)
Back to top