Searched refs:validation_only (Results 1 – 3 of 3) sorted by relevance
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
D | convert_nodes.cc | 263 bool validation_only, in ValidateTensorProperties() argument 294 if (validation_only) return Status::OK(); in ValidateTensorProperties() 1227 const bool validation_only, in PrepareTensorForShape() argument 1244 if (validation_only) { in PrepareTensorForShape() 1654 if (params->validation_only) return Status::OK(); in BinaryTensorOpWeight() 1824 if (params->validation_only) return Status::OK(); in ConvertConv2DHelper() 1970 if (params->validation_only) return Status::OK(); in BinaryTensorOpTensor() 2067 if (params->validation_only) return Status::OK(); in ConvertTranspose() 2166 if (params->validation_only) return Status::OK(); in ConvertReshape() 2214 if (params->validation_only) return Status::OK(); in ConvertExpandDims() [all …]
|
D | convert_nodes.h | 356 validation_only(arg_validation_only), in OpConverterParams() 363 const bool validation_only; member 502 const bool validation_only,
|
D | convert_nodes_test.cc | 488 if (!params->validation_only) start_conversion = true; in TEST_F() 719 for (bool validation_only : {false, true}) { in TEST_F() 723 validation_only, &output_tensor), in TEST_F() 731 tw, GetTestDims({-1, 2}), validation_only, &output_tensor)); in TEST_F() 732 if (validation_only) { in TEST_F() 740 tw, GetTestDims({10, 3}), validation_only, &output_tensor)); in TEST_F() 741 if (validation_only) { in TEST_F() 754 for (bool validation_only : {false, true}) { in TEST_F() 756 tw, GetTestDims({10, 3}), validation_only, &output_tensor)); in TEST_F() 757 if (validation_only) { in TEST_F()
|