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

  /external/tensorflow/tensorflow/core/framework/
node_def_builder_test.cc 63 DataTypeVector in_types, out_types; local
65 InOutTypesForNode(node_def, builder.op_def(), &in_types, &out_types);
71 DataTypeVectorString(out_types));
    [all...]
  /external/tensorflow/tensorflow/core/graph/
gradients.cc 131 DataTypeVector out_types = n->input_types(); local
132 for (int i = 0; i < out_types.size(); ++i) {
133 if (out_types[i] == DT_RESOURCE) {
135 out_types[i] = DT_FLOAT;
138 AddNodeAttr("Tout", out_types, &ndef);
  /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
convert_graph.cc 413 std::vector<DataType> out_types; local
447 out_types.resize(conn.port_number + 1);
450 out_types.at(conn.port_number) = conn.connection_type;
569 .Attr("OutT", out_types)
    [all...]

Completed in 3094 milliseconds