Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for shape_inference_func (0.25 sec)

  1. 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)
  2. 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)
Back to top