Searched refs:lhs_dilate (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | conv_depthwise_test.cc | 36 int64 output_feature, window, stride, pad, lhs_dilate; member
82 config.lhs_dilate = feature / 32; in GetConv2DTestCases()
86 config.stride = config.pad = config.lhs_dilate = -1; in GetConv2DTestCases()
125 absl::StrAppend(&str, "_lhs_dilation_", spec.lhs_dilate, "x1"); in DepthwiseConvolution2DTestDataToString()
208 spec.stride, 0, 0, spec.lhs_dilate, spec.output_feature); in BuildHloTextDepthwiseConvolution2D()
|
| D | grouped_convolution_test.cc | 36 int64 input_feature, output_feature, window, stride, pad, lhs_dilate; member
91 config.lhs_dilate = output_feature / 32; in GetConv2DTestCases()
95 config.stride = config.pad = config.lhs_dilate = -1; in GetConv2DTestCases()
134 absl::StrAppend(&str, "_lhs_dilation_", spec.lhs_dilate, "x1"); in GroupedConvolution2DTestDataToString()
220 spec.stride, 0, 0, spec.lhs_dilate, spec.group_count); in BuildHloTextGroupedConvolution2D()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_parser.cc | 3017 std::vector<int64> lhs_dilate; in ParseWindow() local
3036 return ParseDxD("lhs_dilate", &lhs_dilate); in ParseWindow()
3061 if (!lhs_dilate.empty() && lhs_dilate.size() != size.size()) { in ParseWindow()
3080 lhs_dilate.empty() ? 1 : lhs_dilate[i]); in ParseWindow()
|
| D | hlo_parser_test.cc | 439 …nput, f32[3,3,512,512]{3,2,1,0} %filter), window={size=3x3 pad=1_2x1_2 lhs_dilate=2x2 rhs_reversal… in CreateTestCases()
|