Searched defs:output_feature (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | grouped_convolution_test.cc | 40 int64_t input_feature, output_feature, window, stride, pad, lhs_dilate; member
67 int64_t output_feature = option[0]; in GetConv2DTestCases() local
|
| H A D | convolution_test_1d.cc | 62 int64_t output_feature; member
76 int64_t output_feature = GetParam().output_feature; in TestImpl() local
|
| H A D | conv_depthwise_common.h | 35 int64_t output_feature = -1, window = -1, stride = -1, pad = -1, member
|
| H A D | convolution_dimension_numbers_test.cc | 36 int64_t input_second_spatial, int64_t output_batch, int64_t output_feature, in CreateConvDimensionNumbers()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | convolution_group_converter.cc | 175 int64_t output_feature = filter_shape.dimensions(kernel_output_feature_dim); in GetExpandedFilterMask() local
235 const int64_t output_feature = in HandleBatchGroupCount() local
|
| H A D | hlo_cost_analysis.cc | 602 const int64_t output_feature = in GetConvolutionFlops() local
|
| H A D | elemental_ir_emitter.cc | 3039 llvm::Value* output_feature = index[dnums.output_feature_dimension()]; in EmitConvolution() local
|