Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for PartitionedCall (0.19 sec)

  1. tensorflow/compiler/mlir/tensorflow/transforms/tf_passes.td

        ```
    
        `_XlaMustCompile=true` in the following code
    
        ```mlir
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 21:18:05 UTC 2024
    - 99.6K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc

      // `potential_refined_type`. Return true if the type was changed.
      bool RefineResultType(Operation* op, Value result,
                            Type potential_refined_type);
    
      // Infers the shape from a (Stateful)PartitionedCall operation by looking up
      // the called function and propagating the return type.
      bool InferShapeForCall(CallOpInterface call_op);
    
      bool InferShapeForCast(Operation* op);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Jun 08 07:28:49 UTC 2024
    - 134.1K bytes
    - Viewed (0)
  3. 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;
    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