Searched refs:output_feature_map_count (Results 1 – 4 of 4) sorted by relevance
431 output_feature_map_count(), input_feature_map_count(), in ToString()445 std::string od = absl::StrCat("od", output_feature_map_count()); in ToShortString()471 int64 ret = output_feature_map_count() * input_feature_map_count(); in ComputeWeightCount()
429 int64 bias_count() const { return output_feature_map_count(); } in bias_count()431 int64 output_feature_map_count() const { return tensor_.dimensions(0); } in output_feature_map_count() function
592 dims[0] = filter_descriptor.output_feature_map_count(); in CudnnFilterDescriptor()
723 dims[0] = filter_descriptor.output_feature_map_count(); in ScopedFilterDescriptor()