- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for PartitionedCall (0.28 sec)
-
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) -
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) -
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)