Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for AddNodeAttr (0.34 sec)

  1. tensorflow/compiler/jit/shape_inference.cc

                } else {
                  value.add_int64_val(dim.size());
                }
              }
              AddNodeAttr("value", value, &const_def);
              for (auto const& attr : n->attrs()) {
                if (*attr.first.begin() == '_') {
                  AddNodeAttr(attr.first, attr.second, &const_def);
                }
              }
    
              TF_ASSIGN_OR_RETURN(Node * const_node, graph->AddNode(const_def));
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 31 00:41:19 UTC 2024
    - 13K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/translate/import_model.cc

      }
      // Update op name, drop inputs and set attributes required by the Placeholder
      // op.
      *node->mutable_op() = "Placeholder";
      node->clear_attr();
      node->clear_input();
      AddNodeAttr("dtype", dtype, node);
      AddNodeAttr("shape", it->second.shape, node);
      return absl::OkStatus();
    }
    
    // Preprocesses GraphDef before it can be converted to Graph by,
    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