Searched defs: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
|
| D | convolution_dimension_numbers_test.cc | 39 int64 input_second_spatial, int64 output_batch, int64 output_feature, in CreateConvDimensionNumbers()
|
| D | conv_depthwise_test.cc | 36 int64 output_feature, window, stride, pad, lhs_dilate; member
|
| D | convolution_test.cc | 1719 int64 output_feature; member
1733 int64 output_feature = GetParam().output_feature; in TestImpl() local
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_embedding_output_layout_utils.cc | 48 TPUEmbeddingOutputLayout::FeatureDescriptor* output_feature = in AddDefaultEmbeddingOutputLayoutIfNeeded() local
|
| /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
389 int64 output_feature = in HandleConvolution() local
488 int64 output_feature = in HandleConvolution() local
|
| D | hlo_cost_analysis.cc | 449 const int64 output_feature = in HandleConvolution() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 966 llvm::Value* output_feature = index[dnums.output_feature_dimension()]; in EmitElementalConvolution() local
|