Searched defs:input_feature_dimension (Results 1 – 4 of 4) sorted by relevance
265 int64 input_feature_dimension = backward_conv_dnums.input_feature_dimension(); in MatchBackwardFilter() local563 int64 input_feature_dimension = dnums.kernel_input_feature_dimension(); in MatchBackwardInput() local
132 int64 input_feature_dimension = dim_numbers.input_feature_dimension(); in HandleBackwardFilterBatchGroupConvolution() local
218 const int64 input_feature_dimension = dim_numbers.input_feature_dimension(); in HandleBatchGroupCount() local579 const int64 input_feature_dimension = dim_numbers.input_feature_dimension(); in HandleConvolution() local
497 int64 input_feature_dimension = 8; field