Searched defs:TestParam (Results 1 – 4 of 4) sorted by relevance
72 struct TestParam { struct73 std::vector<std::vector<Tensor>> input_tensors;74 std::vector<Tensor> expected_outputs;75 DataTypeVector expected_output_dtypes;76 std::vector<PartialTensorShape> expected_output_shapes;77 int64 expected_cardinality;78 std::vector<int> breakpoints;
82 struct TestParam { struct83 std::vector<RangeDatasetParam> input_range_dataset_params;84 std::vector<Tensor> expected_outputs;85 std::vector<int> breakpoints;
73 struct TestParam { struct76 explicit TestParam(TestParamTuple param_tuple) in TestParam() function85 static std::string TestNameSuffix( in TestNameSuffix()94 DepthwiseConvImplementation forced_invocation =96 int tests_to_run = 0;97 bool test_stride = false;98 bool test_pad = false;99 bool test_depth_multiplier = false;100 DepthwiseConvOutputRounding output_rounding =102 bool loose_tolerance = false;
27 class TestParam(unittest.TestCase): class