Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for MutableAttrs (0.09 sec)

  1. tensorflow/compiler/mlir/tfr/integration/node_expansion_pass.cc

      LOG_FIRST_N(INFO, 1) << "Run Node Expansion Passes";
    
      // Get the FunctionDef and insert that into the context
      const NodeDef& ndef = orig_op->MutableAttrs()->BuildNodeDef();
      auto& ctx = orig_op->EagerContext();
      Fprint128 cache_key =
          orig_op->MutableAttrs()->CacheKey(orig_op->DeviceName());
      // Include soft placement policy in cache key since the placement strategy
      // can change and thus affect which kernel is picked.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sun Feb 25 16:22:36 UTC 2024
    - 3.8K bytes
    - Viewed (0)
  2. tensorflow/c/eager/c_api.cc

            "Got a null or uninitialized `op` argument");
        return;
      }
      tensorflow::EagerOperation* operation =
          OperationFromInterface(tensorflow::unwrap(const_cast<TFE_Op*>(op)));
      operation->MutableAttrs()->Set(attr_name, attr_value);
    }
    
    TF_CAPI_EXPORT extern int TFE_OpGetInputLength(TFE_Op* op,
                                                   const char* input_name,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 09 08:11:23 UTC 2024
    - 44K bytes
    - Viewed (0)
Back to top