Searched refs:output_feature (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | grouped_convolution_test.cc | 40 int64 input_feature, output_feature, window, stride, pad, lhs_dilate; member
66 int64 output_feature = option[0]; in GetConv2DTestCases() local
77 config.output_feature = output_feature; in GetConv2DTestCases()
84 config.kernel_dims = {kernel_size, kernel_size, group_size, output_feature}; in GetConv2DTestCases()
91 activation_size + kernel_size, output_feature}; in GetConv2DTestCases()
92 } else if (output_feature == 256) { in GetConv2DTestCases()
96 config.lhs_dilate = output_feature / 32; in GetConv2DTestCases()
97 config.output_dims = {batch, output_feature / 32, in GetConv2DTestCases()
98 activation_size - kernel_size + 1, output_feature}; in GetConv2DTestCases()
102 activation_size - kernel_size + 1, output_feature}; in GetConv2DTestCases()
|
| D | convolution_test_1d.cc | 64 int64 output_feature; member
78 int64 output_feature = GetParam().output_feature; in TestImpl() local
85 output_feature}; in TestImpl()
118 std::vector<T> expect_elems(batch * output_feature * num_windows, in TestImpl()
121 auto expected_r3 = expected_r1.Reshape({batch, num_windows, output_feature}) in TestImpl()
|
| D | conv_depthwise_common.cc | 83 spec.window, spec.window, spec.window, spec.output_feature); in BuildHloTextDepthwiseConvolution2D()
107 spec.output_feature); in BuildHloTextDepthwiseConvolution2D()
131 spec.stride, 0, 0, spec.lhs_dilate, spec.output_feature); in BuildHloTextDepthwiseConvolution2D()
|
| D | convolution_dimension_numbers_test.cc | 39 int64 input_second_spatial, int64 output_batch, int64 output_feature, in CreateConvDimensionNumbers() argument
53 dimension_numbers.set_output_feature_dimension(output_feature); in CreateConvDimensionNumbers()
|
| D | conv_depthwise_common.h | 34 int64 output_feature = -1, window = -1, stride = -1, pad = -1, member
|
| D | conv_depthwise_test.cc | 52 config.output_feature = feature; in GetConv2DTestCases()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | convolution_group_converter.cc | 171 int64 output_feature = filter_shape.dimensions(kernel_output_feature_dim); in GetExpandedFilterMask() local
179 GetMaskIds(output_feature / group_count, group_count); in GetExpandedFilterMask()
230 const int64 output_feature = in HandleBatchGroupCount() local
233 if (output_feature != batch_group_count || input_batch != batch_group_count) { in HandleBatchGroupCount()
|
| D | hlo_cost_analysis.cc | 584 const int64 output_feature = in HandleConvolution() local
687 output_feature * in HandleConvolution()
|
| D | elemental_ir_emitter.cc | 2835 llvm::Value* output_feature = index[dnums.output_feature_dimension()]; in EmitConvolution() local
2946 kernel_multi_index[dnums.kernel_output_feature_dimension()] = output_feature; in EmitConvolution()
|
| /external/llvm-project/mlir/docs/Rationale/ |
| D | Rationale.md | 683 // kernel: [kernel_height, kernel_width, input_feature, output_feature]
684 // output: [batch, output_height, output_width, output_feature]
691 affine.for %of = 0 to %output_feature {
|