Searched refs:output_feature (Results 1 – 8 of 8) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/tests/ |
D | grouped_convolution_test.cc | 36 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()
|
D | conv_depthwise_test.cc | 36 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()
|
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 | convolution_test.cc | 1719 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/ |
D | convolution_group_converter.cc | 172 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()
|
D | hlo_cost_analysis.cc | 449 const int64 output_feature = in HandleConvolution() local 528 output_feature * in HandleConvolution()
|
/external/tensorflow/tensorflow/core/tpu/ |
D | tpu_embedding_output_layout_utils.cc | 48 TPUEmbeddingOutputLayout::FeatureDescriptor* output_feature = in AddDefaultEmbeddingOutputLayoutIfNeeded() local 51 output_feature->add_output_location(); in AddDefaultEmbeddingOutputLayoutIfNeeded()
|
/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
D | ir_emitter.cc | 966 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()
|