Searched refs:input_feature (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | convolution_test_1d.cc | 63 int64 input_feature; member
77 int64 input_feature = GetParam().input_feature; in TestImpl() local
83 input_feature}; in TestImpl()
84 std::vector<int64> filter_dims = {window_size, input_feature, in TestImpl()
119 static_cast<T>(window_size * input_feature)); in TestImpl()
|
| D | grouped_convolution_test.cc | 40 int64 input_feature, output_feature, window, stride, pad, lhs_dilate; member
70 int64 input_feature = option[4]; in GetConv2DTestCases() local
76 config.group_count = input_feature / group_size; in GetConv2DTestCases()
81 input_feature}; in GetConv2DTestCases()
|
| D | convolution_dimension_numbers_test.cc | 38 int64 input_batch, int64 input_feature, int64 input_first_spatial, in CreateConvDimensionNumbers() argument
45 dimension_numbers.set_input_feature_dimension(input_feature); in CreateConvDimensionNumbers()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_conv_rewriter.cc | 80 int64 input_feature = lhs->shape().dimensions(input_feature_dimension); in ConvertBatchGroupedToFeatureGroupedConvolution() local
110 new_shape.set_dimensions(input_feature_dimension, input_feature * num_groups); in ConvertBatchGroupedToFeatureGroupedConvolution()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | conv_op_helpers.cc | 53 int64 input_feature = filter_shape.dimensions(input_feature_dim); in GroupedFilterShapeForDepthwiseConvolution() local
59 depthwise_multiplier * input_feature); in GroupedFilterShapeForDepthwiseConvolution()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_cost_analysis.cc | 582 const int64 input_feature = in HandleConvolution() local
686 const int64 fma_count = (input_feature / convolution->feature_group_count()) * in HandleConvolution()
|
| D | dynamic_dimension_inference.cc | 275 HloInstruction* input_feature = parent_->GetDynamicSize( in HandleCustomCall() local
280 if (input_feature != nullptr && kernel_feature == nullptr) { in HandleCustomCall()
|
| D | elemental_ir_emitter.cc | 2861 llvm::Value* input_feature = in EmitConvolution() local
2933 input_multi_index[dnums.input_feature_dimension()] = input_feature; in EmitConvolution()
2945 kernel_multi_index[dnums.kernel_input_feature_dimension()] = input_feature; in EmitConvolution()
|
| /external/llvm-project/mlir/docs/Rationale/ |
| D | Rationale.md | 682 // input: [batch, input_height, input_width, input_feature]
683 // kernel: [kernel_height, kernel_width, input_feature, output_feature]
694 affine.for %if = 0 to %input_feature {
|