Home
last modified time | relevance | path

Searched refs:activation_dims (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dconv_depthwise_test.cc37 std::vector<int64> activation_dims; member
68 config.activation_dims = {batch, activation_size, activation_size, feature}; in GetConv2DTestCases()
117 "activation_dims_", absl::StrJoin(spec.activation_dims, "x"), in DepthwiseConvolution2DTestDataToString()
136 if (spec.activation_dims[1] == 1 && spec.kernel_dims[1] == 2) { in BuildHloTextDepthwiseConvolution2D()
149 data_type, absl::StrJoin(spec.activation_dims, ","), in BuildHloTextDepthwiseConvolution2D()
155 absl::StrJoin(spec.activation_dims, ","), in BuildHloTextDepthwiseConvolution2D()
174 data_type, absl::StrJoin(spec.activation_dims, ","), in BuildHloTextDepthwiseConvolution2D()
180 absl::StrJoin(spec.activation_dims, ","), in BuildHloTextDepthwiseConvolution2D()
198 data_type, absl::StrJoin(spec.activation_dims, ","), in BuildHloTextDepthwiseConvolution2D()
204 absl::StrJoin(spec.activation_dims, ","), in BuildHloTextDepthwiseConvolution2D()
Dgrouped_convolution_test.cc38 std::vector<int64> activation_dims; member
76 config.activation_dims = {batch, activation_size, activation_size, in GetConv2DTestCases()
126 "activation_dims_", absl::StrJoin(spec.activation_dims, "x"), in GroupedConvolution2DTestDataToString()
145 if (spec.activation_dims[1] == 1 && spec.kernel_dims[1] == 2) { in BuildHloTextGroupedConvolution2D()
159 data_type, absl::StrJoin(spec.activation_dims, ","), in BuildHloTextGroupedConvolution2D()
165 absl::StrJoin(spec.activation_dims, ","), in BuildHloTextGroupedConvolution2D()
185 data_type, absl::StrJoin(spec.activation_dims, ","), in BuildHloTextGroupedConvolution2D()
191 absl::StrJoin(spec.activation_dims, ","), in BuildHloTextGroupedConvolution2D()
210 data_type, absl::StrJoin(spec.activation_dims, ","), in BuildHloTextGroupedConvolution2D()
216 absl::StrJoin(spec.activation_dims, ","), in BuildHloTextGroupedConvolution2D()
Dconv_depthwise_backprop_filter_test.cc37 std::vector<int64> activation_dims; member
67 config.activation_dims = {batch, activation_size, activation_size, feature}; in GetConv2DTestCases()
88 config.activation_dims = {batch, activation_size, activation_size, in GetConv2DTestCases()
114 "activation_dims_", absl::StrJoin(spec.activation_dims, "x"), in BatchGroupedConvolution2DTestDataToString()
141 data_type, absl::StrJoin(spec.activation_dims, ","), in BuildHloTextBatchGroupedConvolution2D()
147 absl::StrJoin(spec.activation_dims, ","), in BuildHloTextBatchGroupedConvolution2D()