Home
last modified time | relevance | path

Searched refs:output_feature (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dgrouped_convolution_test.cc36 int64 input_feature, output_feature, window, stride, pad, lhs_dilate; member
62 int64 output_feature = option[0]; in GetConv2DTestCases() local
73 config.output_feature = output_feature; in GetConv2DTestCases()
80 config.kernel_dims = {kernel_size, kernel_size, group_size, output_feature}; in GetConv2DTestCases()
86 activation_size + kernel_size, output_feature}; in GetConv2DTestCases()
87 } else if (output_feature == 256) { in GetConv2DTestCases()
91 config.lhs_dilate = output_feature / 32; in GetConv2DTestCases()
92 config.output_dims = {batch, output_feature / 32, in GetConv2DTestCases()
93 activation_size - kernel_size + 1, output_feature}; in GetConv2DTestCases()
97 activation_size - kernel_size + 1, output_feature}; in GetConv2DTestCases()
Dconv_depthwise_test.cc36 int64 output_feature, window, stride, pad, lhs_dilate; member
65 config.output_feature = feature; in GetConv2DTestCases()
159 spec.window, spec.window, spec.window, spec.output_feature); in BuildHloTextDepthwiseConvolution2D()
184 spec.output_feature); in BuildHloTextDepthwiseConvolution2D()
208 spec.stride, 0, 0, spec.lhs_dilate, spec.output_feature); in BuildHloTextDepthwiseConvolution2D()
Dconvolution_dimension_numbers_test.cc39 int64 input_second_spatial, int64 output_batch, int64 output_feature, in CreateConvDimensionNumbers() argument
53 dimension_numbers.set_output_feature_dimension(output_feature); in CreateConvDimensionNumbers()
Dconvolution_test.cc1719 int64 output_feature; member
1733 int64 output_feature = GetParam().output_feature; in TestImpl() local
1740 output_feature}; in TestImpl()
1773 std::vector<T> expect_elems(batch * output_feature * num_windows, in TestImpl()
1776 auto expected_r3 = expected_r1.Reshape({batch, num_windows, output_feature}) in TestImpl()
/external/tensorflow/tensorflow/compiler/xla/service/
Dconvolution_group_converter.cc172 int64 output_feature = filter_shape.dimensions(kernel_output_feature_dim); in GetExpandedFilterMask() local
180 GetMaskIds(output_feature / group_count, group_count); in GetExpandedFilterMask()
389 int64 output_feature = in HandleConvolution() local
401 if (group_count == output_feature && !filter_expansion_) { in HandleConvolution()
488 int64 output_feature = in HandleConvolution() local
491 int64 filter_slice_width = output_feature / group_count; in HandleConvolution()
Dhlo_cost_analysis.cc449 const int64 output_feature = in HandleConvolution() local
528 output_feature * in HandleConvolution()
/external/tensorflow/tensorflow/core/tpu/
Dtpu_embedding_output_layout_utils.cc48 TPUEmbeddingOutputLayout::FeatureDescriptor* output_feature = in AddDefaultEmbeddingOutputLayoutIfNeeded() local
51 output_feature->add_output_location(); in AddDefaultEmbeddingOutputLayoutIfNeeded()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc966 llvm::Value* output_feature = index[dnums.output_feature_dimension()]; in EmitElementalConvolution() local
1075 kernel_multi_index[dnums.kernel_output_feature_dimension()] = output_feature; in EmitElementalConvolution()