Searched refs:GetInOutTypes (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | insert_logging.cc | 114 TF_RETURN_IF_ERROR(GetInOutTypes(node, &input_types, &output_types)); in InsertLogging()
|
| D | transform_utils.h | 133 Status GetInOutTypes(const NodeDef& node_def, DataTypeVector* inputs,
|
| D | transform_utils_test.cc | 822 TF_EXPECT_OK(GetInOutTypes(*float_const_def, &float_const_inputs, in TestGetInOutTypes()
832 GetInOutTypes(*int_const_def, &int_const_inputs, &int_const_outputs)); in TestGetInOutTypes()
840 TF_EXPECT_OK(GetInOutTypes(*float_relu_def, &float_relu_inputs, in TestGetInOutTypes()
851 GetInOutTypes(*int_relu_def, &int_relu_inputs, &int_relu_outputs)); in TestGetInOutTypes()
|
| D | transform_utils.cc | 587 Status GetInOutTypes(const NodeDef& node_def, DataTypeVector* inputs, in GetInOutTypes() function
|
| D | quantize_nodes.cc | 705 GetInOutTypes(float_node, &input_types, &output_types)); in QuantizeNodes()
|