Searched refs:input_feature_map_count (Results 1 – 4 of 4) sorted by relevance
431 output_feature_map_count(), input_feature_map_count(), in ToString()446 std::string id = absl::StrCat("id", input_feature_map_count()); in ToShortString()471 int64 ret = output_feature_map_count() * input_feature_map_count(); in ComputeWeightCount()
432 int64 input_feature_map_count() const { return tensor_.dimensions(1); } in input_feature_map_count() function
593 dims[1] = filter_descriptor.input_feature_map_count(); in CudnnFilterDescriptor()
724 dims[1] = filter_descriptor.input_feature_map_count(); in ScopedFilterDescriptor()