- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for FunctionOpInterface (0.18 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc
int M = f.getNumArguments() - N; DCOMMENT_OP(op, "Inferring shape for with N = " << N << " and M = " << M); // Initialize with function input types. auto input_types = llvm::to_vector<1>( cast<FunctionOpInterface>(f.getOperation()).getArgumentTypes()); DatasetInput input_elements = GetDatasetInput(op->getOperand(0)); if (!input_elements) { op->emitWarning("unexpected dataset input; skipping function refinement");
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jun 08 07:28:49 UTC 2024 - 134.1K bytes - Viewed (0)