Home
last modified time | relevance | path

Searched defs:input_feature (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dgrouped_convolution_test.cc36 int64 input_feature, output_feature, window, stride, pad, lhs_dilate; member
66 int64 input_feature = option[4]; in GetConv2DTestCases() local
Dconvolution_dimension_numbers_test.cc38 int64 input_batch, int64 input_feature, int64 input_first_spatial, in CreateConvDimensionNumbers()
Dconvolution_test.cc1718 int64 input_feature; member
1732 int64 input_feature = GetParam().input_feature; in TestImpl() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dconv_op_helpers.cc133 int64 input_feature = filter_shape.dimensions(input_feature_dim); in ReshapeFilterForDepthwiseConvolution() local
/external/tensorflow/tensorflow/contrib/tensor_forest/kernels/
Dtree_utils.cc27 int32 input_feature, const tensorforest::TensorForestDataSpec& spec) { in FindDenseFeatureSpec()
32 int32 input_feature, const tensorforest::TensorForestDataSpec& spec) { in FindSparseFeatureSpec()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_cost_analysis.cc447 const int64 input_feature = in HandleConvolution() local
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc990 llvm::Value* input_feature = in EmitElementalConvolution() local