- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for hasAttr (0.08 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc
return changed; } bool ShapeInference::InferShapeForFunctionAttachedToXlaHostCompute( XlaHostComputeOp op) { const std::string kShapeInferenceGraph = "shape_inference_graph"; if (!op->hasAttr(kShapeInferenceGraph)) { return false; } ModuleOp module = op->getParentOfType<ModuleOp>(); func::FuncOp func = module.lookupSymbol<func::FuncOp>(
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
// attribute. This provides a hook for the user to represent the variable // tensor in the MLIR level. if (auto* inst = value.getDefiningOp(); inst && inst->hasAttr("tfl.is_variable")) { is_variable = true; } bool has_rank = type.hasRank(); if (shape_signature.empty()) { return tflite::CreateTensor(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:41:49 UTC 2024 - 164.5K bytes - Viewed (0)