HomeSort by relevance Sort by last modified time
    Searched defs:SetNodeAttr (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/tools/graph_transforms/
transform_utils.h 76 inline void SetNodeAttr(const string& key, const T& value, NodeDef* node) {
88 SetNodeAttr(key, tensor_proto, node);
  /external/tensorflow/tensorflow/core/grappler/clusters/
single_machine_test.cc 379 inline void SetNodeAttr(const string& key, const T& value, NodeDef* node) {
386 inline void SetNodeAttr(const string& key, const Tensor& tensor,
390 SetNodeAttr(key, tensor_proto, node);
404 SetNodeAttr("key_dtype", key_dtype, hashtable_node);
405 SetNodeAttr("value_dtype", data_dtype, hashtable_node);
411 SetNodeAttr("dtype", key_dtype, keys_node);
415 SetNodeAttr("value", keys, keys_node);
420 SetNodeAttr("dtype", data_dtype, values_node);
424 SetNodeAttr("value", values, values_node);
430 SetNodeAttr("Tkey", key_dtype, init_table_node)
    [all...]

Completed in 185 milliseconds