Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for DataTypeIsInteger (0.2 sec)

  1. tensorflow/c/experimental/gradients/math_grad.cc

                           AbstractTensorHandle** output, const char* name) {
      auto dtype = input->DataType();
      if (DataTypeIsFloating(BaseType(dtype)) ||
          DataTypeIsInteger(BaseType(dtype))) {
        return tensorflow::ops::Identity(ctx, input, output, name);
      } else if (!DataTypeIsComplex(BaseType(dtype)) &&
                 BaseType(dtype) != DT_VARIANT) {
        return errors::InvalidArgument(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 28 13:53:47 UTC 2024
    - 15.2K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/mark_for_compilation_pass.cc

      }
    
      if (n->type_string() != "AssignAddVariableOp" &&
          n->type_string() != "AssignSubVariableOp") {
        return false;
      }
    
      DataType dtype;
      if (!TryGetNodeAttr(n->def(), "dtype", &dtype) || !DataTypeIsInteger(dtype)) {
        return false;
      }
    
      Node* const_input = nullptr;
      for (const Edge* e : n->in_edges()) {
        if (!e->IsControlEdge() && e->src()->IsConstant()) {
          const_input = e->src();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 21 12:19:41 UTC 2024
    - 85.3K bytes
    - Viewed (0)
Back to top