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

  /external/tensorflow/tensorflow/compiler/tf2xla/
tf2xla_util.cc 167 auto& attr_map = *d->mutable_attr(); local
168 attr_map["dtype"].set_type(info.data_type);
169 *attr_map["shape"].mutable_shape() = info.feed->shape();
  /external/tensorflow/tensorflow/core/example/
example_parser_configuration.cc 57 auto& attr_map = node.attr(); local
58 auto num_sparse = attr_map.at("Nsparse").i();
59 auto num_dense = attr_map.at("Ndense").i();
63 auto tdense = attr_map.at("Tdense");
64 auto dense_shapes = attr_map.at("dense_shapes");
65 auto sparse_types = attr_map.at("sparse_types");
  /external/tensorflow/tensorflow/tools/graph_transforms/
transform_utils.h 79 auto* attr_map = node->mutable_attr(); local
80 (*attr_map)[key] = attr_value;
  /external/tensorflow/tensorflow/core/grappler/clusters/
single_machine_test.cc 373 auto* attr_map = node->mutable_attr(); local
374 (*attr_map)[key] = attr_value;

Completed in 135 milliseconds