Search Options

Display Count
Sort
Preferred Language
Advanced Search

Results 1 - 2 of 2 for tensor_dtypes (0.06 seconds)

  1. tensorflow/c/eager/gradients.cc

      std::vector<int64_t> tensor_ids(tensors.size());
      std::vector<tensorflow::DataType> tensor_dtypes(tensors.size());
      for (int i = 0; i < tensors.size(); i++) {
        tensor_ids[i] = ToId(tensors[i]);
        tensor_dtypes[i] = tensors[i]->DataType();
      }
      return GradientTape::ShouldRecord(tensor_ids, tensor_dtypes);
    }
    void Tape::DeleteTrace(const AbstractTensorHandle* t) {
      GradientTape::DeleteTrace(ToId(t));
    }
    Created: Tue Apr 07 12:39:13 GMT 2026
    - Last Modified: Tue Feb 24 06:18:31 GMT 2026
    - 19.6K bytes
    - Click Count (0)
  2. tensorflow/c/eager/c_api_test_util.cc

      TF_Status* status = TF_NewStatus();
      TFE_Op* op = TFE_NewOp(ctx, op_name.c_str(), status);
      CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
      TF_DeleteStatus(status);
    
      TFE_OpSetAttrType(op, "tensor_type", TF_INT32);
      TFE_OpSetAttrString(op, "tensor_name", "dummy", 5);
      TFE_OpSetAttrString(op, "send_device", send_device.c_str(),
                          send_device.size());
    Created: Tue Apr 07 12:39:13 GMT 2026
    - Last Modified: Thu Oct 09 05:56:18 GMT 2025
    - 23.4K bytes
    - Click Count (0)
Back to Top