Searched refs:DepthwiseConvolution2DSpec (Results 1 – 1 of 1) sorted by relevance
35 struct DepthwiseConvolution2DSpec { struct48 ::testing::tuple<DepthwiseConvolution2DSpec, bool>> {};50 static std::vector<DepthwiseConvolution2DSpec> GetConv2DTestCases() { in GetConv2DTestCases()51 std::vector<DepthwiseConvolution2DSpec> config_set; in GetConv2DTestCases()64 DepthwiseConvolution2DSpec config; in GetConv2DTestCases()113 ::testing::tuple<DepthwiseConvolution2DSpec, bool>>& data) { in DepthwiseConvolution2DTestDataToString()134 const DepthwiseConvolution2DSpec& spec, bool use_bfloat16) { in BuildHloTextDepthwiseConvolution2D()213 const DepthwiseConvolution2DSpec& spec = ::testing::get<0>(GetParam()); in XLA_TEST_P()