Searched refs:input_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
66 int64 input_feature = option[4]; in GetConv2DTestCases() local
72 config.group_count = input_feature / group_size; in GetConv2DTestCases()
77 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()
|
| D | convolution_test.cc | 1718 int64 input_feature; member
1732 int64 input_feature = GetParam().input_feature; in TestImpl() local
1738 input_feature}; in TestImpl()
1739 std::vector<int64> filter_dims = {window_size, input_feature, in TestImpl()
1774 static_cast<T>(window_size * input_feature)); in TestImpl()
|
| /external/tensorflow/tensorflow/contrib/tensor_forest/kernels/ |
| D | tree_utils.cc | 27 int32 input_feature, const tensorforest::TensorForestDataSpec& spec) { in FindDenseFeatureSpec() argument
28 return static_cast<DataColumnTypes>(spec.GetDenseFeatureType(input_feature)); in FindDenseFeatureSpec()
32 int32 input_feature, const tensorforest::TensorForestDataSpec& spec) { in FindSparseFeatureSpec() argument
37 while (input_feature >= size_sum && column_num < spec.sparse_size()) { in FindSparseFeatureSpec()
|
| D | tree_utils.h | 70 int32 input_feature, const tensorforest::TensorForestDataSpec& spec);
72 int32 input_feature, const tensorforest::TensorForestDataSpec& spec);
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | conv_op_helpers.cc | 133 int64 input_feature = filter_shape.dimensions(input_feature_dim); in ReshapeFilterForDepthwiseConvolution() local
139 output_feature_dim, depthwise_multiplier * input_feature); in ReshapeFilterForDepthwiseConvolution()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_cost_analysis.cc | 447 const int64 input_feature = in HandleConvolution() local
527 const int64 fma_count = (input_feature / convolution->feature_group_count()) * in HandleConvolution()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 990 llvm::Value* input_feature = in EmitElementalConvolution() local
1062 input_multi_index[dnums.input_feature_dimension()] = input_feature; in EmitElementalConvolution()
1074 kernel_multi_index[dnums.kernel_input_feature_dimension()] = input_feature; in EmitElementalConvolution()
|