- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for shape_inference_func (0.25 sec)
-
tensorflow/c/c_api_experimental.cc
if (!status->status.ok()) return; if (op_reg_data->shape_inference_fn == nullptr) { status->status = InvalidArgument("No shape inference function exists for op '", node_def.op(), "', did you forget to define it?"); return; } status->status = c.Run(op_reg_data->shape_inference_fn); if (!status->status.ok()) return; // Set output_shapes.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 15 03:35:10 UTC 2024 - 29.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/import_model.cc
if (!op_reg_data) { DVLOG(1) << "Skipping inference for unregistered op " << node.type_string(); return default_type(); } if (op_reg_data->shape_inference_fn == nullptr) { DVLOG(1) << "Skipping inference for op without shape function " << node.type_string(); return default_type(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 183.2K bytes - Viewed (0)